Skip to content

Ding1119/Mask-RCNN_TB

Repository files navigation

Tuberculosis detection using Mask-RCNN

This is an example of object detection on Micro bacterium tuberculosis using Mask-RCNN.

This is an implementation of Mask R-CNN on Python 3, Keras, and TensorFlow. The model generates bounding boxes and segmentation masks for each instance of an object in the image. It's based on Feature Pyramid Network (FPN) and a ResNet101 backbone.

Instance Segmentation Sample Instance Segmentation Sample

The repository includes:

  • Source code of Mask R-CNN built on FPN and ResNet101.
  • Training code for MS COCO
  • Pre-trained weights for MS COCO
  • Jupyter notebooks to visualize the detection pipeline at every step
  • ParallelModel class for multi-GPU training
  • Evaluation on MS COCO metrics (AP)
  • Example of training on your own dataset

About

This is an example of object detection on Micro bacterium tuberculosis using Mask-RCNN

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published