search

D641593 / MixNet

MIT License
60 stars 9 forks source link

I need your help.. + custom train #11

Open Hwaaan2 opened 9 months ago

Hwaaan2 commented 9 months ago

Thank you for your good work.

Can you tell me in detail how to do custom train?

It hasn't been long since I studied AI, so I'm asking for help due to lack of knowledge.

Lastly, I wonder what is the file 'poly_gt_DJI_0013_6930.0.JPG.mat' and where can I download the file.

Thank you.

D641593 commented 9 months ago

Hi, the .mat file is the format of my ground-truth file. In my dataloader code, I assume that each image (.jpg) has a corresponding ground-truth file (.mat) with the same file name.

The file "poly_gt_DJI_0013_6930.0.JPG.mat" is requested because your input image name is "poly_gt_DJI_0013_6930.0.JPG".

For custom datasets, you must convert your ground-truth file to match the default format in my dataloader. There are many dataloader files because each dataset has a different ground truth format. Maybe you could create a dataloader file for your custom dataset and override the part that reads the ground-truth file.