matterport / Mask_RCNN

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

mask_rcnn multiple classes #1059

Open ssetty opened 5 years ago

ssetty commented 5 years ago

What changes in source code needs to be done to implement multiple classes? Default source code is only for single class. Kindly help. Thanks

hate-deadline commented 5 years ago

@ssetty you can add many class as you want, by using add_class() and add_image().

ssetty commented 5 years ago

Thanks. I guess we need to change methods loadmask(), load method - add_image() what needs to be changed?

KenKhor commented 5 years ago

Duplicate of #372