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

Classify music genre from a 10 second sound stream using a Neural Network.

This is the codebase for Diffusion Models Beat GANS on Image Synthesis.

Segmentation vgg16 fcn - cityscapes

Fastshap: A fast, approximate shap kernel

Code of Classification Saliency-Based Rule for Visible and Infrared Image Fusion

[CVPRW 21] "BNN - BN = ? Training Binary Neural Networks without Batch Normalization", Tianlong Chen, Zhenyu Zhang, Xu Ouyang, Zechun Liu, Zhiqiang Shen, Zhangyang Wang

Nerf pl - NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning

Official Python implementation of the 'Sparse deconvolution'-v0.3.0

UniMoCo: Unsupervised, Semi-Supervised and Full-Supervised Visual Representation Learning

Implementation of the paper Recurrent Glimpse-based Decoder for Detection with Transformer.

Pytorch implementation of AREL

PyTorch implementation of DARDet: A Dense Anchor-free Rotated Object Detector in Aerial Images

AI4Good project for detecting waste in the environment

A curated (most recent) list of resources for Learning with Noisy Labels

A cross-document event and entity coreference resolution system, trained and evaluated on the ECB+ corpus.

PIXIE: Collaborative Regression of Expressive Bodies

Negative Interactions for Improved Collaborative Filtering:

PyTorch implementation of the Quasi-Recurrent Neural Network - up to 16 times faster than NVIDIA's cuDNN LSTM

Randomizes the warps in a stock pokeemerald repo.

Code & Models for Temporal Segment Networks (TSN) in ECCV 2016