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

OOD Generalization and Detection (ACL 2020)

Train Yolov4 using NBX-Jobs

Official code for "Towards An End-to-End Framework for Flow-Guided Video Inpainting" (CVPR2022)

Python with OpenCV - MediaPip Framework Hand Detection

An efficient PyTorch implementation of the winning entry of the 2017 VQA Challenge.

A TensorFlow implementation of the Mnemonic Descent Method.

PyTorch implementations of the beta divergence loss.

A simple pygame dino game which can also be trained and played by a NEAT KI

Trans-Encoder: Unsupervised sentence-pair modelling through self- and mutual-distillations

Anomaly detection related books, papers, videos, and toolboxes

Deep Learning and Reinforcement Learning Library for Scientists and Engineers 🔥

Implementation of U-Net and SegNet for building segmentation

A GridMixup augmentation, inspired by GridMask and CutMix

Official implementation of the ICML2021 paper "Elastic Graph Neural Networks"

This repo is the official implementation for Multi-Scale Adaptive Graph Neural Network for Multivariate Time Series Forecasting

The implementation of PEMP in paper "Prior-Enhanced Few-Shot Segmentation with Meta-Prototypes"

Pytorch implementation of "Get To The Point: Summarization with Pointer-Generator Networks"

Deep Q-Learning Network in pytorch (not actively maintained)

Pathdreamer: A World Model for Indoor Navigation

Categorical Depth Distribution Network for Monocular 3D Object Detection