Dataset Preparation

[LoveIsAGame]

Thank you for your work!
I try to use the new data set for training, but I don't know how to construct my data format. Can you help me?

Look forward to your reply！

[LoveIsAGame]

Can I use my own data set to construct data in coco or VOC format for training? Because this is a weakly supervised network, I don't know how it uses the data in it.

[shenyunhang]

Yes, we can use our own data set to construct data in coco format for training.

To use custom datasets, please refer to this documents.

We also recommend to use pycococreator to generate annotations in the format COCO requires.

[LoveIsAGame]

Thanks,i will try. And I have another problem ，The screenshot is below。

[shenyunhang]

Hi, this problem is not directly related to DRN-WSOD-pytorch.

It should be some problem with PyTorch and CUDA.