cvlab-stonybrook / LearningToCountEverything

MIT License
357 stars 72 forks source link

Training on new dataset #10

Closed Muhibullah1 closed 3 years ago

Muhibullah1 commented 3 years ago

Hi, I tried to train your model on my own dataset. I labeled images on CVAT (you recommended) but the annotations are in XML format and you coded for .json file. I was wondering how did you create json file out of XML, although I manually changed it to json but doest not working. I will be thankful for your guidance

Viresh-R commented 3 years ago

Hey, I used an Xml parser [1] to extract the annotations from CVAT XML files, copied all the annotations into a python dict and saved the dict as a json file.

[1]https://docs.python.org/3/library/xml.etree.elementtree.html

Muhibullah1 commented 3 years ago

Hey, I used an Xml parser [1] to extract the annotations from CVAT XML files, copied all the annotations into a python dict and saved the dict as a json file.

[1]https://docs.python.org/3/library/xml.etree.elementtree.html

Thank you for the quick response . I will look at that