Streamlit tool to explore coco datasets

Last update: Dec 16, 2022

Overview

What is this

This tool given a COCO annotations file and COCO predictions file will let you explore your dataset, visualize results and calculate important metrics.

Running the explorer on example data

You can use the predictions I prepared and explore the results on the COCO validation dataset. The predictions are coming from a Mask R-CNN model trained with mmdetection.

Download (and extract in project directory) the labels, annotations and images:

https://drive.google.com/open?id=1wxIagenNdCt_qphEe8gZYK7H2_to9QXl

Setup using docker

sudo docker run -p 8501:8501 -it -v "$PWD"/coco_data:/coco_data i008/cocoexp:latest  \
    --coco_train /coco_data/ground_truth_annotations.json \
    --coco_predictions /coco_data/predictions.json  \
    --images_path /coco_data/images/

Setup using conda

conda env update
conda activate cocoexplorer
streamlit run coco_explorer.py -- --coco_train ./coco_data/ground_truth_annotations.json --coco_predictions ./coco_data/predictions.json  --images_path ./coco_data/val2017/

Setup using pip

python3 -m venv .venv
. .venv/bin/activate
pip install -r requirements.txt
streamlit run coco_explorer.py -- --coco_train ./coco_data/ground_truth_annotations.json --coco_predictions ./coco_data/predictions.json  --images_path ./coco_data/val2017/

go to http://localhost:8501

Running on your own data

In the same way you can explore your own results. Just follow the official COCO dataset format for annotations and predictions.

Streamlit tool to explore coco datasets

Related tags

Overview

What is this

Running the explorer on example data

Running on your own data

Examples

Owner

Jakub Cieslik

A collection of awesome resources image-to-image translation.

Fast EMD for Python: a wrapper for Pele and Werman's C++ implementation of the Earth Mover's Distance metric

In this tutorial, you will perform inference across 10 well-known pre-trained object detectors and fine-tune on a custom dataset. Design and train your own object detector.

This Jupyter notebook shows one way to implement a simple first-order low-pass filter on sampled data in discrete time.

A PyTorch implementation of "SelfGNN: Self-supervised Graph Neural Networks without explicit negative sampling"

Breaching - Breaching privacy in federated learning scenarios for vision and text

Official PyTorch implementation of UACANet: Uncertainty Aware Context Attention for Polyp Segmentation

Hummingbird compiles trained ML models into tensor computation for faster inference.

Pytorch implementation for the paper: Contrastive Learning for Cold-start Recommendation

LONG-TERM SERIES FORECASTING WITH QUERYSELECTOR – EFFICIENT MODEL OF SPARSEATTENTION

Differential fuzzing for the masses!

A novel framework to automatically learn high-quality scanning of non-planar, complex anisotropic appearance.

Adversarial examples to the new ConvNeXt architecture

Official Keras Implementation for UNet++ in IEEE Transactions on Medical Imaging and DLMIA 2018

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

The code for paper Efficiently Solve the Max-cut Problem via a Quantum Qubit Rotation Algorithm

Tightness-aware Evaluation Protocol for Scene Text Detection

Causal Influence Detection for Improving Efficiency in Reinforcement Learning

Code for weakly supervised segmentation of a single class

PyTorch Implementation of ByteDance's Cross-speaker Emotion Transfer Based on Speaker Condition Layer Normalization and Semi-Supervised Training in Text-To-Speech