https://github.com/vanvalenlab/deepcell-retinamask

RetinaNet and RetinaMask models for object detection based on TensorFlow and DeepCell-tf.
https://github.com/vanvalenlab/deepcell-retinamask

maskrcnn object-detection retinanet tensorflow

Last synced: 8 months ago
JSON representation

RetinaNet and RetinaMask models for object detection based on TensorFlow and DeepCell-tf.

• Host: GitHub
• URL: https://github.com/vanvalenlab/deepcell-retinamask
• Owner: vanvalenlab
• License: other
• Created: 2020-06-30T00:30:12.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2021-11-29T17:43:11.000Z (almost 4 years ago)
• Last Synced: 2025-02-04T23:06:55.343Z (8 months ago)
• Topics: maskrcnn, object-detection, retinanet, tensorflow
• Language: Python
• Homepage:
• Size: 2.05 MB
• Stars: 1
• Watchers: 2
• Forks: 0
• Open Issues: 2
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# DeepCell-RetinaMask

[![Build Status](https://github.com/vanvalenlab/deepcell-retinamask/workflows/build/badge.svg)](https://github.com/vanvalenlab/deepcell-retinamask/actions)

[![Coverage Status](https://coveralls.io/repos/github/vanvalenlab/deepcell-retinamask/badge.svg?branch=master)](https://coveralls.io/github/vanvalenlab/deepcell-retinamask?branch=master)

[![Apache 2.0](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://github.com/vanvalenlab/deepcell-retinamask/blob/master/LICENSE)

[![PyPI version](https://badge.fury.io/py/DeepCell-RetinaMask.svg)](https://badge.fury.io/py/deepcell-retinamask)

[![Python Versions](https://img.shields.io/pypi/pyversions/deepcell-retinamask.svg)](https://pypi.org/project/deepcell-retinamask/)

`deepcell-retinamask` is a deep learning library for building RetinaNet and RetinaMask based object detection models with `tensorflow` and `deepcell-tf`.

This project was heavily influenced by [keras-retinanet](https://github.com/fizyr/keras-retinanet) and [keras-maskrcnn](https://github.com/fizyr/keras-maskrcnn).

## Install

`deepcell-retinamask` can be easily installed with pip:

```bash

$ pip install deepcell-retinamask

```

## Examples

For examples of how to train models with the `deepcell-retinamask` library, check out the following notebooks:

- [Training a RetinaNet Model](https://github.com/vanvalenlab/deepcell-retinamask/blob/master/notebooks/RetinaNet.ipynb)

- [Training a RetinaMask Model](https://github.com/vanvalenlab/deepcell-retinamask/blob/master/notebooks/RetinaMask.ipynb)

- [Training a PanOpticFPN Model](https://github.com/vanvalenlab/deepcell-retinamask/blob/master/notebooks/PanOpticFPN.ipynb)