yxgeee / MMT

[ICLR-2020] Mutual Mean-Teaching: Pseudo Label Refinery for Unsupervised Domain Adaptation on Person Re-identification.
https://yxgeee.github.io/projects/mmt
MIT License
473 stars 72 forks source link

How to include more dataset ? #19

Closed absagargupta closed 4 years ago

absagargupta commented 4 years ago

I am currently working and have got results on Sysu Dataset(through some other algorithm). What changes do I make in the code so as to run this on Sysu. I have made Sysu.py just like dukemtmc.py and placed the dataset at the same place as other dataset. I have also included it in init.py (__factory) of the datasets folder. Whenever I try running the sh scripts/pretrain.sh Sysu market1501 resnet50 1 it show the error that only defaults datasets "dukemtmc", "msmt", market1501 can be used.

yxgeee commented 4 years ago

Did you run python setup.py install after changing the code?

absagargupta commented 4 years ago

Ohh. Thank you. I forgot to run it again. So any changes I make in the original code I have to do that python setup.py install. Got it :-)