scaelles / DEXTR-PyTorch

Deep Extreme Cut http://www.vision.ee.ethz.ch/~cvlsegmentation/dextr
GNU General Public License v3.0
843 stars 153 forks source link

I have a some question about pascal voc train #8

Closed dypark86 closed 6 years ago

dypark86 commented 6 years ago

Thanks for your jobs!

I have a some question about train method I train pascal voc dataset model with basic setting. and evaluate model with eval_all.py.

but there are some gap in the performance.

there are some more training technic?

thanks!

kmaninis commented 6 years ago

Hi,

Could you provide some more info about the setup you are using? Python/PyTorch version, GPU, etc.? The number should be reproducible just by running the script, without changing any parameters (tripple checked).

dypark86 commented 6 years ago

i tried with following setup

after setting the data path i edit just following code -> before: device = torch.device("cuda:"+str(gpu_id) if torch.cuda.is_available() else "cpu") -> after: device = torch.device("cuda") because setgpuid make error in my env

and i train with train_pascal.py and eval with eval_all.py

Thanks!

scaelles commented 6 years ago

Hello, We tried with the same configuration except that we are using a TItan Xp, but it shouldn't make a difference. Could you try to modify the threshold in eval_all.py? Best, Sergi