fcjian / InstaGen

InstaGen: Enhancing Object Detection by Training on Synthetic Dataset, CVPR2024
https://fcjian.github.io/InstaGen
MIT License
68 stars 2 forks source link

Issues with coco_crop dataset when running the model #8

Closed sunny1401 closed 2 months ago

sunny1401 commented 2 months ago

Hello - I have been trying to run the model provided here. However, the batch is always none at collation. Upon inspection - I saw that you have this line. However there is no definition of prepare_data in either the class itself or the parent class. As much as I know, PyTorch dataset class doesn't have any such function. Could you please tell what am I missing? Also could you tell what is name_mapping