ADE20K_To_COCO

KO | EN

gitlite — search

#typescript #ai-agents #ai #dsh-plugin #deepseek-harness #open-source #claude-code #codex #cli #developer-tools #react #windows

★ 15 Open GitHub ↗

A tool to convert ADE20K to COCO format based on desired objects

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# ADE20K_To_COCO
This tool can convert the latest [ADE20K dataset](https://groups.csail.mit.edu/vision/datasets/ADE20K/) to COCO format, which can be further used in Detectron2 Framework.
As my goal is to build a object level SLAM system, this tool is designed to retrieve information related to target objects.
However, you can always modify it to fit your purposes, and Welcome to pull a request.

Please :star: the repo if you find it useful :full_moon_with_face:

## Usage
1. Install dependencies
    ```
    pip install -r requirements.txt
    ```
2. Run Script
    ```
    python AdeToCOCO.py --datasetDir DATASETDIR --pklPath PKLPATH --objectNames OBJECTNAMES [OBJECTNAMES ...]
    ```
    objectNames: Name of target objects, ex: ["door", "door frame"]\
    pklPath: the path to index_ade20k.pkl provied by ADE20K\
    datasetDir: ADE20K dataset path

## Test Result
Use demo.ipynb to visualize if it works\
![image](https://raw.githubusercontent.com/macTracyHuang/ADE20K_To_COCO/main/figure/door1.png)\
![image](https://raw.githubusercontent.com/macTracyHuang/ADE20K_To_COCO/main/figure/wall1.png)\

## Citation
```
@misc{
    title  = {ADE20K_To_COCO},
    author = {Tracy Huang},
    url    = {https://github.com/macTracyHuang/ADE20K_To_COCO},
    year   = {2023}
}
```

ADE20K_To_COCO

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README