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

CondenseNet: Light weighted CNN for mobile devices

The official implementation of paper Siamese Transformer Pyramid Networks for Real-Time UAV Tracking, accepted by WACV22

This is the repo for our work "Towards Persona-Based Empathetic Conversational Models" (EMNLP 2020)

Generating images from caption and vice versa via CLIP-Guided Generative Latent Space Search

Weakly- and Semi-Supervised Panoptic Segmentation (ECCV18)

Code for Environment Inference for Invariant Learning (ICML 2020 UDL Workshop Paper)

SGoLAM - Simultaneous Goal Localization and Mapping

[Machine Learning Engineer Basic Guide] 부스트캠프 AI Tech - Product Serving 자료

Code for the paper "Reinforced Active Learning for Image Segmentation"

Learning from History: Modeling Temporal Knowledge Graphs with Sequential Copy-Generation Networks

An implementation of a discriminant function over a normal distribution to help classify datasets.

Multi-Task Deep Neural Networks for Natural Language Understanding

using yolox+deepsort for object-tracker

This project aims at building a real-time wide band channel sounder using USRPs

An official reimplementation of the method described in the INTERSPEECH 2021 paper - Speech Resynthesis from Discrete Disentangled Self-Supervised Representations.

Self-supervised Multi-modal Hybrid Fusion Network for Brain Tumor Segmentation

AntroPy: entropy and complexity of (EEG) time-series in Python

A neuroanatomy-based augmented reality experience powered by computer vision. Features 3D visuals of the Atlas Brain Map slices.

2021 CCF BDCI 全国信息检索挑战杯（CCIR-Cup）智能人机交互自然语言理解赛道第二名参赛解决方案

novel deep learning research works with PaddlePaddle