Skip to content

SSD polyp detection tool using MobileNet as feature extractor.

License

Notifications You must be signed in to change notification settings

hackobi/polyp-detection-tool

Repository files navigation

Polyp Detection Demo

Usage

In a browser open and run the polyp detection tool Colab Notebook.

Requirements

The required packages will be installed in the Colab notebook

Using a frozen TensorFlow graph to run inference locally

Step 1: Download frozen_inference_graph.pband label_map.pbtxt

If you want to run inference on your local system you'll need the frozen inference graph and label map that you get after training.

Feel free to try these which work well on polyp detection.

Step 2:

Run locally

local_inference_test.ipynb