Repository files navigation

Please downloadADE dataset and make sure the file structures are as follow:

ADE20K|-images/|-index_ade20k.mat|-index_ad220k.pkl|-objects.txt

You need to runpython3 edit_dataset.py to generate data list for ADE20K dataset, after that the file structures will be:

ADE20K|-images/|-datalist/|-index_ade20k.mat|-index_ad220k.pkl|-objects.txt

Please download MIT imagenet pretrainedresnet101-imagenet.pth and store it atpretrain/resnet101-imagenet.pth, then runpython3 edit_pretrain_resnet.py to getpretrain/resnet101_base.pth for Jittor reloading.

Please download pytorch version pretrainedvan_b2.pth and store it atpretrain/van_b2.pth, then runpython3 edit_pretrain_van.py to getpretrain/van_b2_base.pth for Jittor reloading.

In order to trainxxx model (xxx is eitherresnet orvan), please run the following instruction, and the output log will be stored atlog/xxx/yyyymmdd_hhmmss.log

./train_xxx.sh

In case of machine crash or remote disconnection, we recommend that you usetmux to execute training or testing.