• Stars
    star
    112
  • Rank 312,240 (Top 7 %)
  • Language
    Python
  • Created 11 months ago
  • Updated 4 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Official Code for Tracking Any Object Amodally

TAO-Amodal

Official Repository of Tracking Any Object Amodally.

๐Ÿ“™ Project Page | ๐Ÿ“Ž Paper Link | โœ๏ธ Citations

TAO-Amodal

๐Ÿ“Œ Leave a โญ to keep track of our updates.


Table of Contents


๐ŸŽ’ Get Started

Clone the repository

git clone https://github.com/WesleyHsieh0806/TAO-Amodal.git 

Setup environment

conda create --name TAO-Amodal python=3.9 -y
conda activate TAO-Amodal
bash environment_setup.sh

๐Ÿ“š Prepare Dataset

  1. Download our dataset following the instructions here.
  2. The directory should have the following structure:
    TAO-Amodal
     โ”œโ”€โ”€ frames
     โ”‚    โ””โ”€โ”€ train
     โ”‚       โ”œโ”€โ”€ ArgoVerse
     โ”‚       โ”œโ”€โ”€ BDD
     โ”‚       โ”œโ”€โ”€ Charades
     โ”‚       โ”œโ”€โ”€ HACS
     โ”‚       โ”œโ”€โ”€ LaSOT
     โ”‚       โ””โ”€โ”€ YFCC100M
     โ”œโ”€โ”€ amodal_annotations
     โ”‚    โ”œโ”€โ”€ train/validation/test.json
     โ”‚    โ”œโ”€โ”€ train_lvis_v1.json
     โ”‚    โ””โ”€โ”€ validation_lvis_v1.json
     โ”œโ”€โ”€ example_output
     โ”‚    โ””โ”€โ”€ prediction.json
     โ”œโ”€โ”€ BURST_annotations
     โ”‚    โ”œโ”€โ”€ train
     โ”‚         โ””โ”€โ”€ train_visibility.json
     โ”‚    ...

Explore more examples from our dataset here.

๐Ÿง‘โ€๐ŸŽจ Visualization

Visualize our dataset and tracker predictions to get a better understanding of amodal tracking. Instructions could be found here.

TAO-Amodal

๐Ÿƒ Training and Inference

We provide the training and inference code of the proposed Amodal Expander.

The inference code generates a lvis_instances_results.json, which could be used to obtain the evaluation results as introduced in the next section.

๐Ÿ“Š Evaluation

  1. Output tracker predictions as json. The predictions should be structured as:
[{
    "image_id" : int,
    "category_id" : int,
    "bbox" : [x,y,width,height],
    "score" : float,
    "track_id": int,
    "video_id": int
}]

We also provided an example output prediction json here. Refer to this file to check the correct format.

  1. Evaluate on TAO-Amodal
cd tools
python eval_on_tao_amodal.py --track_result /path/to/prediction.json \
                             --output_log   /path/to/output.log \
                             --annotation   /path/to/validation_lvis_v1.json

Annotation JSON is provided in our dataset. Evaluation results will be written in your console and saved in --output_log.

Citations

@misc{hsieh2023tracking,
    title={Tracking Any Object Amodally},
    author={Cheng-Yen Hsieh and Tarasha Khurana and Achal Dave and Deva Ramanan},
    year={2023},
    eprint={2312.12433},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}

More Repositories

1

SS-PRL

SS-PRL: Self-Supervised Pyramid Representation Learning for Multi-Label Visual Analysis (IEEE WACV 2023)
Shell
87
star
2

C3-SL

C3-SL: Circular Convolution-Based Batch-Wise Compression for Communication-Efficient Split Learning (IEEE MLSP 2022)
Python
70
star
3

FL-HDC

FL-HDC: Hyperdimensional Computing Design for the Application of Federated Learning (IEEE AICAS2021)
Python
68
star
4

Downstair-Battle

An interesting two-player version of NFShaft with designed characters and ult.
C
67
star
5

Amodal-Expander

Official Training and Inference Code of Amodal Expander, Proposed in Tracking Any Object Amodally
Jupyter Notebook
13
star
6

Wes-ML-Transfer-Learning

Transfer Learning of hand drawn images with real images
Python
10
star
7

Wes-ML-Anomaly-Detection

Image Anomaly Detection with AutoEncoder
Python
9
star
8

Wes-ML-CNN

CNN Classification-Food Recognition
Python
9
star
9

Wes-ML-RNN

Text Sentiment Classification
Python
9
star
10

Wes-ML-Seq-to-Seq

Sequence to Sequence with RNN
Python
8
star
11

Wes-ML-Network-Compression

Compress the model size with several techniques
Python
8
star
12

Wes-ML-Explainable-AI

Explainable AI-Observation of output of CNN Layers
Python
8
star
13

CS231n-Assignments

CS231n Assignment implementation
Jupyter Notebook
8
star
14

HDC

Wes's ML Project ---HDC(Hyper Dimensional Computing)
Python
8
star
15

DSP-Assignment

DSP Assignments
Python
8
star
16

Wes-ML-Regression

Linear Regression Model
Python
8
star
17

Data-structure-PA

่ณ‡ๆ–™็ตๆง‹(้กๆ•™ๆŽˆ) PA HW
C++
7
star
18

hello

This is a repository for my code practice
Python
7
star
19

Wes-Algorithm

Algorithm implementation
6
star
20

Wes-ML-Classification

ML Classification Task
Python
6
star
21

Machine-Learning-Model

Repository for model release
6
star
22

Leetcode_practice

Leetcode็ทด็ฟ’้กŒไธŠๅ‚ณ
Python
6
star
23

Algorithm-PA

ๆผ”็ฎ—ๆณ•PAไฝœๆฅญ
C++
6
star
24

C-practice

This is for C++ practice
C++
6
star
25

Open-Category-Mesh-Prediction

A Generalized Mesh Prediction Framework for Arbitrary Objects | CMU Learning for 3D Vision 16825 | Team 7 Project
Jupyter Notebook
1
star