pytorch-deeplab-xception icon indicating copy to clipboard operation
pytorch-deeplab-xception copied to clipboard

Can I train it on the CoCo Panoptic dataset?

Open squashking opened this issue 5 years ago • 5 comments

Sorry for the newbie question. I see there is train_coco.sh file, I am just wondering if it can be used to train the Coco Panoptic dataset. If so, how should I organize the train data (i.e. the directory structure of the dataset)? Thanks a lot!

squashking avatar Nov 15 '19 08:11 squashking

I have not seen the blocks (instance segmentation head, instance segmentation losses) needed for the instance segmentation task but you could maybe add them yourself

RubenS02 avatar Nov 16 '19 17:11 RubenS02

@RubenS02 Thanks. That's too challenging for me since I'm a beginner. I'd go for semantic segmentation instead. Can you please tell me how to organize the train data? I mean, using the Coco 2017 train dataset. Thank you!

squashking avatar Nov 17 '19 23:11 squashking

@squashking I have not trained with COCO, but everything you need is in dataloaders/datasets/coco.py

RubenS02 avatar Nov 18 '19 08:11 RubenS02

I am also a novice, can I train my own data set? I also do semantic segmentation, I need to manually make the data set, how should I use this network training, and what should I modify?

kungkook avatar Mar 09 '20 07:03 kungkook

Please dont nectropost. Making your own dataset is a LOT of work. You can look into tools like labelme for that. Then, use a dataset format like coco style, and modify the coco.py dataset to suit your own.

faheuer avatar Mar 10 '20 14:03 faheuer