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

The pyrelational package offers a flexible workflow to enable active learning with as little change to the models and datasets as possible

Galileo library for large scale graph training by JD

PyTorch implementation of "PatchGame: Learning to Signal Mid-level Patches in Referential Games" to appear in NeurIPS 2021

This repo uses a combination of logits and feature distillation method to teach the PSPNet model of ResNet18 backbone with the PSPNet model of ResNet50 backbone. All the models are trained and tested on the PASCAL-VOC2012 dataset.

Generalized Data Weighting via Class-level Gradient Manipulation

Learning a mapping from images to psychological similarity spaces with neural networks.

YOLOV4运行在嵌入式设备上

Pytorch Implementation for Dilated Continuous Random Field

The mini-MusicNet dataset

MagFace: A Universal Representation for Face Recognition and Quality Assessment

Generating Videos with Scene Dynamics

Experiments on Flood Segmentation on Sentinel-1 SAR Imagery with Cyclical Pseudo Labeling and Noisy Student Training

Easily Process a Batch of Cox Models

Drone Task1 - Drone Task1 With Python

Implementation of paper "DCS-Net: Deep Complex Subtractive Neural Network for Monaural Speech Enhancement"

House-GAN++: Generative Adversarial Layout Refinement Network towards Intelligent Computational Agent for Professional Architects

Very Deep Convolutional Networks for Large-Scale Image Recognition

Lucid library adapted for PyTorch

ROS Basics and TurtleSim

Learning Representational Invariances for Data-Efficient Action Recognition