DRN-WSOD-pytorch icon indicating copy to clipboard operation
DRN-WSOD-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon shenyunhang

Dataset Preparation

Open LoveIsAGame opened this issue 3 years ago • 4 comments

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 avatar Dec 27 '20 06:12 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.

LoveIsAGame avatar Dec 27 '20 07:12 LoveIsAGame

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.

shenyunhang avatar Dec 27 '20 17:12 shenyunhang

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

LoveIsAGame avatar Dec 28 '20 02:12 LoveIsAGame

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

It should be some problem with PyTorch and CUDA.

shenyunhang avatar Dec 28 '20 02:12 shenyunhang