/ritm_annotation

Tool to annotate semantic segmentation datasets

Primary LanguagePythonMIT LicenseMIT

ritm_annotation

codecov CI

Tool to do dataset annotation for semantic segmentation datasets.

Based on SamsungLabs/ritm_interactive_segmentation

Work in progress

Installation

  • Pip
pip install ritm_annotation
  • Nix flakes

Running it using Nix flakes

nix run github:lucasew/ritm_annotation -- --help
  • Docker
docker run -ti ghcr.io/lucasew/ritm_annotation:latest bash

Usage

$ python -m ritm_annotation --help
# or
$ ritm_annotation --help

Pretrained model weights

Here you can download pretrained weights

In my tests I used coco_lvis_h18_itermask.pth (SHA256 5f69cfce354d1507e3850bfc39ee7057c8dd27b6a4910d1d2dc724916b9ee32b)

Development

Read the CONTRIBUTING.md file.