search

cmb-chula / pylon

Official implementation of Pyramid Localization Network (PYLON).
Apache License 2.0
25 stars 9 forks source link

new datasets #3

Open xiaoli-10086 opened 2 years ago

xiaoli-10086 commented 2 years ago

how to run the experiments with my own new datasets? (my datasets is covid-19 CT images)

phizaz commented 2 years ago

A good starting point is to ponder the differences between train_nih.py (Chest X-Ray 14 dataset) and train_voc.py (Pascal VOC dataset). These files rely on data/nih.py and data/voc.py which you can take a look. This is the right way to do if you are intended to use the same code base.

However, if you are accustomed to training a deep model by yourself. There is nothing stopping you from just utilizing the PYLON model pylon.py with your own training loop. This is the easy way potentially removes a lot of boilerplates.