search

GuoleiSun / CountSeg

Official code for "Object counting and instance segmentation with image-level supervision", in CVPR 2019 and TPAMI 2020
162 stars 43 forks source link

How to train my custom dataset? #4

Closed Wilbur529 closed 5 years ago

Wilbur529 commented 5 years ago

Hi, Thanks for your reply in issues#3:) Now, i attempt to train CountSeg with my custom dataset, Could you please give me some advice about it?

Wilbur529 commented 5 years ago

Done. Reimplemented a custom version of this function: https://github.com/GuoleiSun/CountSeg/blob/51cde8a6644ccfd8594d47b57fd7b24538372a7f/Nest-pytorch/install/datasets.py#L231.