MengyangPu / EDTER

EDTER: Edge Detection with Transformer, in CVPR 2022
MIT License
280 stars 34 forks source link

Poor metrics for training the nyud-hha dataset #80

Closed liuxingyu123 closed 1 year ago

liuxingyu123 commented 1 year ago

Hello, I used configuration file of EDTER_BIMLA_320x320_40k_nyud_hha_bs_4.py directly when training the nyud hha dataset. After training, the test is conducted with ods=0.640. May I ask what parameters need to be changed during training?

MengyangPu commented 1 year ago

After testing NYUD-HHA-StageI, we get ODS=0.690, OIS=0.707. We upload all files to BaiDuNetdisk. You can change the parameters following the training logs. Thanks~

MengyangPu commented 1 year ago

NYUD-HHA-StageII can be download from here.