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

Automatic differentiation with weighted finite-state transducers.

a Pytorch easy re-implement of "YOLOX: Exceeding YOLO Series in 2021"

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.

Official Implementation of DE-DETR and DELA-DETR in "Towards Data-Efficient Detection Transformers"

CycleTransGAN-EVC: A CycleGAN-based Emotional Voice Conversion Model with Transformer

Code for unmixing audio signals in four different stems "drums, bass, vocals, others". The code is adapted from "Jukebox: A Generative Model for Music"

Pytorch implementation of few-shot semantic image synthesis

PyGCL: Graph Contrastive Learning Library for PyTorch

A collection of metrics for evaluating timbre dissimilarity using the TorchMetrics API

A library that can print Python objects in human readable format

Ensemble Learning Priors Driven Deep Unfolding for Scalable Snapshot Compressive Imaging [PyTorch]

Supervised domain-agnostic prediction framework for probabilistic modelling

Generative Handwriting using LSTM Mixture Density Network with TensorFlow

"Learning and Analyzing Generation Order for Undirected Sequence Models" in Findings of EMNLP, 2021

LSUN Dataset Documentation and Demo Code

[SIGGRAPH 2021 Asia] DeepVecFont: Synthesizing High-quality Vector Fonts via Dual-modality Learning

Object-Centric Learning with Slot Attention

Split Variational AutoEncoder

Pytorch Implementation of Interaction Networks for Learning about Objects, Relations and Physics

An implementation of "Learning human behaviors from motion capture by adversarial imitation"