clovaai / wsolevaluation

Evaluating Weakly Supervised Object Localization Methods Right (CVPR 2020)
MIT License
332 stars 55 forks source link

Custom datasets #17

Closed santoshreddy254 closed 4 years ago

santoshreddy254 commented 4 years ago

Cloud you throw some light on how to modify the code for other custom datasets? I want to use my own dataset instead of using ImageNet or CUB or OpenImages. I have dataset in VOC format

coallaoh commented 4 years ago

Thank you for your interest in our work. To add a dataset, prepare the metadata mimicking the file structures and file formats in e.g. https://github.com/clovaai/wsolevaluation/tree/master/metadata/CUB.

You then need to adapt the code to this new dataset by changing e.g.

and set the data root (https://github.com/clovaai/wsolevaluation/blob/6692538bc2e396f6076253e814aebea92b12e064/config.py#L136) to your local dataset location.

santoshreddy254 commented 4 years ago

Very quick response Thank you for that. I will update you if I can do that

santoshreddy254 commented 4 years ago

Hi, I am facing issues in how to load VOC format data, you have bounding boxes in a txt file for each image but for me, I have in XML file which is parsed and got the box but I was not able to incorporate in this code. Any help or suggestion is appreciated.

coallaoh commented 4 years ago

Sorry, we don't have enough manpower to provide guidance on turning XML files into txt files :(

coallaoh commented 4 years ago

Closing the issue, assuming the question was answered :) Please re-open the issue as necessary.