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 static analysis library for computing graph representations of Python programs suitable for use with graph neural networks.

G-NIA model from "Single Node Injection Attack against Graph Neural Networks" (CIKM 2021)

🚀 An end-to-end ML applications using PyTorch, W&B, FastAPI, Docker, Streamlit and Heroku

MIRACLE (Missing data Imputation Refinement And Causal LEarning)

Source code of the paper Meta-learning with an Adaptive Task Scheduler.

Convert openmmlab (not only mmdetection) series model to tensorrt

Human head pose estimation using Keras over TensorFlow.

A collection of papers about Transformer in the field of medical image analysis.

Omniscient Video Super-Resolution

Pytorch Implementation of "Contrastive Representation Learning for Exemplar-Guided Paraphrase Generation"

Pytorch implementation of OCNet series and SegFix.

Simple converter for deploying Stable-Baselines3 model to TFLite and/or Coral

Codes and models of NeurIPS2021 paper - DominoSearch: Find layer-wise fine-grained N:M sparse schemes from dense neural networks

Nvdiffrast - Modular Primitives for High-Performance Differentiable Rendering

Unofficial Alias-Free GAN implementation. Based on rosinality's version with expanded training and inference options.

This repository contains the code for the CVPR 2021 paper "GIRAFFE: Representing Scenes as Compositional Generative Neural Feature Fields"

PyTorch implementations for our SIGGRAPH 2021 paper: Editable Free-viewpoint Video Using a Layered Neural Representation.

A Dynamic Residual Self-Attention Network for Lightweight Single Image Super-Resolution

PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)

Demystifying How Self-Supervised Features Improve Training from Noisy Labels