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

CALVIN - A benchmark for Language-Conditioned Policy Learning for Long-Horizon Robot Manipulation Tasks

tensorflow implementation of 'YOLO : Real-Time Object Detection'

Weight estimation in CT by multi atlas techniques

Reverse engineering Rosetta 2 in M1 Mac

The first dataset on shadow generation for the foreground object in real-world scenes.

Official repo for AutoInt: Automatic Integration for Fast Neural Volume Rendering in CVPR 2021

Diffgram - Supervised Learning Data Platform

Convex optimization for fun and profit.

A simple configurable bot for sending arXiv article alert by mail

Official PyTorch code for Mutual Affine Network for Spatially Variant Kernel Estimation in Blind Image Super-Resolution (MANet, ICCV2021)

Code & Data for Enhancing Photorealism Enhancement

Official implementation of Protected Attribute Suppression System, ICCV 2021

PantheonRL is a package for training and testing multi-agent reinforcement learning environments.

A hyperparameter optimization framework

A Large-Scale Dataset for Spinal Vertebrae Segmentation in Computed Tomography

This Repostory contains the pretrained DTLN-aec model for real-time acoustic echo cancellation.

AFLFast (extends AFL with Power Schedules)

Official PyTorch code for the paper: "Point-Based Modeling of Human Clothing" (ICCV 2021)

Selective Wavelet Attention Learning for Single Image Deraining

Implementation of the GBST block from the Charformer paper, in Pytorch