sputummrcnn 0.9

Custom Mask RCNN

A Custom Python Package for Implementing MaskRCNN. This is a private package intended for implementing Mask RCNN Algorithm on Sputum Project, its ML Model generation and related resources.

Installing

Note

It is advised to install in a virtual enviroment

run : pip install sputummrcnn

Features

• Mask RCNN Implementation on Sputum Files.

  Given the set of images and the annotation files the package trains the models as per user requirements. The Images can then be predicted or cropped using the other features in the package.

• Feature Extraction.

  Based on the Cropped Images, certain features are extracted by the package, and then saved as a parquet file which can be used as training dataset for the Machine Learning model.

• Machine Learning Model & Analysis

  After Feature files are generated, based on Random Forest or K Nearest Neighbour Algorithm a Machine Learning Model is trained based on user preference. It’s accuracy and confusion is also shown so as to select the best one.

Credits

https://github.com/matterport/Mask_RCNN