ZhangYuanhan-AI/Bamboo

Yuanhan Zhang¹  Qinghong Sun²  Yichun Zhou³  Zexin He³  Zhenfei Yin⁴  Kun Wang⁴ 
Lv Sheng³  Yu Qiao⁵  Jing Shao⁴  Ziwei Liu¹

¹S-Lab, Nanyang Technological University  ²Beijing University of Posts and Telecommunication 
³Beihang University  ⁴SenseTime Research  ⁵Shanghai AI Laboratory

TL;DR

Bamboo is a mega-scale and information-dense dataset for classification and detection pre-training. It is built upon integrating 24 public datasets (e.g. ImagenNet, Places365, Object365, OpenImages) and added new annotations through active learning. Bamboo has 69M image classification annotations (4 times larger than ImageNet) and 32M object bounding boxes (2 times larger than Object365).

---

[Paper]• [Project]

Leaderboard

🥳!
🥳!

Updates

[11/2022] We release Bamboo-Det.
[10/2022] We won the first place in Computer Vision in the Wild Challenge(ImageNet-1K in Pre-training track). 🥳!
[06/2022] We split Bamboo-CLS into 30 datasets that represent different realms (e.g. car, mammals, food and etc.) in the natural worlds: HERE
[06/2022] Try out the web demo on Huggingface spaces 🤗: .
[06/2022] We release Bamboo-CLS with FC layer, it can classify 115,217 categories.
[06/2022] We release our label system with many useful attributes!.
[03/2022] Bamboo-CLS ResNet-50 and Bamboo-CLS ViT B/16 have been released.
[03/2022] arXiv paper has been released.

About Bamboo

Explore

Website Link

Downloads

• Sign up an account at HERE, get the USERNAME for opengvlab.
• Send your request to [email protected]. The request should include your name, username and orgnization as follows. We will notify you by email as soon as possible.

  USERNANE: USERNANE(from step1)
  NAME: XXX
  ORGANIZATION: XXX (Bamboo is only for academic research and non-commercial use)
  

Label sytem

We provide the hierarchy for our label system at HERE. This JSON file includes the following attrubutes of each concept. We hope this information will be beneficial for your research.

We take concept/class dog as an example.

• Load JSON file

  #input
  with open('PATH-TO-JSON-FILE.json') as f:
  bamboo = json.load(f)
  print(bamboo.keys())
  

  #output
  'father2child', 'child2father', 'id2name', 'id2desc', 'id2desc_zh', 'id2name_zh'
  

• Check the id (n02084071) of the dog on HERE.
• Get the attrubutes you need.
  • Hypernyms bamboo['child2father']['n02084071']: domestic_animals, canine.
  • Hyponyms bamboo['father2child']['n02084071']: husky, griffon, shiba inu and etc.
  • Description bamboo['id2desc']['n02084071']: a member of the genus Canis (probably descended from the common wolf) that has been domesticated by man since prehistoric times; occurs in many breeds.
  • Included in which public dataset bamboo['id2state']['n02084071']['academic']: openimage, iWildCam2020, STL10, cifar10, iNat2021, ImageNet21K, coco, OpenImage, object365.

Meta File

• Class Name
• Class id -> id
• id -> Class id

Special meta file

Downloading the whole dataset might be unnecessary for most purposes. We provide meta files based on the following dimension.

• Class-wise (e.g. dog, car, boat and etc.)
• Superclass-wise (e.g. animal, transportation, structure and etc.): HERE

How to download files from Google drives in the terminal?

• Install gdown

  pip install gdown
  

• get the id of the files Link, e.g. https://drive.google.com/file/d/1WEKQ_68Y9i9FzakvPYU6Yj5SOvkZCIEm/view?usp=sharing
  id: 1WEKQ_68Y9i9FzakvPYU6Yj5SOvkZCIEm
• Download

  gdown https://drive.google.com/uc?id=1WEKQ_68Y9i9FzakvPYU6Yj5SOvkZCIEm
  

Model Zoo

Bamboo-CLS

Bamboo-DET

Getting Started

Installation

# Create conda environment
conda create -n bamboo python=3.7
conda activate bamboo

# Install Pytorch
conda install pytorch==1.8.0 torchvision==0.9.0 cudatoolkit=10.2 -c pytorch

# Clone and install
git clone https://github.com/Davidzhangyuanhan/Bamboo.git

Linear Probe

Step 1:

Downloading and organizing each downstream dataset as follows

data
├── flowers
│   ├── train/
│   ├── test/
│   ├── train_meta.list
│   ├── test_meta.list

Step 2:

Changing root and meta in Bamboo-Benchmark/configs/100p/config_\*.yaml

Step 3:

Writing the path of the downloaded/your model config in Bamboo-Benchmark/configs/models_cfg/\*.yaml

Step 4:

Writing the name of the downloaded/your model in Bamboo-Benchmark/multi_run_100p.sh

Step 5:

sh Bamboo-Benchmark/multi_run_100p.sh

Citation

If you use this code in your research, please kindly cite the following papers.

@misc{zhang2022bamboo,
      title={Bamboo: Building Mega-Scale Vision Dataset Continually with Human-Machine Synergy}, 
      author={Yuanhan Zhang and Qinghong Sun and Yichun Zhou and Zexin He and Zhenfei Yin and Kun Wang and Lu Sheng and Yu Qiao and Jing Shao and Ziwei Liu},
      year={2022},
      eprint={2203.07845},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Acknowledgement

Thanks to Siyu Chen (https://github.com/Siyu-C) for implementing the Bamboo-Benchmark.