cocosynth

合成画像を使ってcocoモデルを作れるよってツール。

環境と必要なもの

Ubuntu18.04
Anaconda
背景画像
(切り抜いた)物体画像

使い方

git colne https://github.com/akTwelve/cocosynthしてくる。

Getting Startedを見るとSetup方法が書いてある。

# Linux
cd 
git colne https://github.com/akTwelve/cocosynth
conda create -n cocosynth python=3.6
source activate cocosynth
conda install -c conda-forge shapely
cd cocosynth
pip install -r requirements.txt

ディレクトリ構造

.
├── datasets
│   └── README.md
├── docs
│   └── getting-started.md
├── LICENSE
├── notebooks
│   ├── coco_image_viewer.ipynb
│   └── train_mask_rcnn.ipynb
├── python
│   ├── coco_json_utils.py
│   └── image_composition.py
├── README.md
└── requirements.txt

4 directories, 9 files

物体画像と背景画像

画像についてはスマホまたは(Web)カメラなどで撮影しておいてください。

物体画像

認識したい物体をいろんな角度から撮影。枚数は任意です。
撮影できたらGIMPなどで物体のみ切り抜きをおこなってください。
切り抜きが終わればdatasets内に任意のファイルを作成し(今回はobjectsとした)、その中にinputファイルを作成して、各物体ごと(例：books,boxes,bottles)に名前をつけたファイルを用意してその中に保存してください。 3の例のように作成するとこんな感じになっているはずです。

.
├── datasets
│   ├── objects
│   │   └── input
│   │       ├── books
│   │       ├── bottles
│   │       └── boxes
│   └── README.md
├── docs
│   └── getting-started.md
├── LICENSE
├── notebooks
│   ├── coco_image_viewer.ipynb
│   └── train_mask_rcnn.ipynb
├── python
│   ├── coco_json_utils.py
│   └── image_composition.py
├── README.md
└── requirements.txt

9 directories, 9 files

合成画像作成

cd ~/cocosynth/
source activate cocosynth
python ./python/image_composition.py --input_dir ./datasets/objects/input --output_dir ./datasets/objects/output --count 100 --width 512 --height 512