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

Source code for From Stars to Subgraphs

Structured Edge Detection Toolbox

Reference PyTorch implementation of "End-to-end optimized image compression with competition of prior distributions"

Scalable and Elastic Deep Reinforcement Learning Using PyTorch. Please star. 🔥

GeneGAN: Learning Object Transfiguration and Attribute Subspace from Unpaired Data

Lighting the Darkness in the Deep Learning Era: A Survey, An Online Platform, A New Dataset

Semi-automated OpenVINO benchmark_app with variable parameters

InterFaceGAN - Interpreting the Latent Space of GANs for Semantic Face Editing

Change Detection in SAR Images Based on Multiscale Capsule Network

Introduction to AI assignment 1 HCM University of Technology, term 211

An implementation demo of the ICLR 2021 paper Neural Attention Distillation: Erasing Backdoor Triggers from Deep Neural Networks in PyTorch.

Leaf: Multiple-Choice Question Generation

Marine debris detection with commercial satellite imagery and deep learning.

A Simple LSTM-Based Solution for "Heartbeat Signal Classification and Prediction" in Tianchi

This repository contains an implementation of the Permutohedral Attention Module in Pytorch

This is the first released system towards complex meters` detection and recognition, which is implemented by computer vision techniques.

Code for "PV-RAFT: Point-Voxel Correlation Fields for Scene Flow Estimation of Point Clouds", CVPR 2021

Image-retrieval-baseline - MUGE Multimodal Retrieval Baseline

Semi-supervised Implicit Scene Completion from Sparse LiDAR

DeepLab is a state-of-art deep learning system for semantic image segmentation built on top of Caffe.