Testing on custom datasets

[Gokulnath-Sourirajan]

Hi, after having downloaded the pretrained model, is there a way to test them on my custom dataset or ant other datasets that do not have its corresponding .py file in demo folder? Thanks!

[Aditya-5895]

put images inside input_images folder give the path of downloded model in demo/***det.py and run demo/det.py. output will be saved in test_icdar file

[ipheiman]

Hi there, I have followed Aditya's instructions. But when i run "python demo/icdar2015_detection.py", it takes quite awhile (10-15 minutes) before generating an error "RuntimeError: cuDNN error: CUDNN_STATUS_MAPPING_ERROR".

I thought it could be due to the batch size? So I changed the IMS_PER_BATCH in configs yml from 4 to 1, but the same error popped up. Have you encountered this?

[Aditya-5895]

I encountered That error but after reducing batch size it worked. Its a cuda bug .If IMS_PER_BATCH=1 is also not working for you check without cuda or check proper pytorch or cuda version for your GPU. check with this link https://github.com/pytorch/pytorch/issues/27588

[ipheiman]

Thanks Aditya for the quick reply. You're probably right that its a cuda bug, and my GPU is RTX3080 which could be incompatible with cuda 10. Cheers!

[kaxapatel]

Can I use SVHN dataset on this model?