matterport / Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
Other
24.66k stars 11.71k forks source link

Train with VOC-2007 Dataset #883

Open akorez opened 6 years ago

akorez commented 6 years ago

Hi, my dataset format VOC-2007 and i have 6 class. How can i convert my own dataset to COCO format or Can i train Mask RCNN with my own VOC-2007 dataset? Please help me. Thanks...

LackesLab commented 6 years ago

You can train mask r-cnn with your own dataset. You need to bring your data in the correct format. I suggest to learn and understand the train_shapes notebook, which is in the samples folder.

genausz commented 5 years ago

Check out my implementation of TRAINING Mask_RCNN with PASCAL VOC 2012 dataset.

https://github.com/genausz/Mask_RCNN

code are in dir ./samples/voc

might several changes need to be made for VOC 2007.