File size: 750 Bytes
fadb92b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# create COCO-style dataset for CubiCasa5k
python -m data_preprocess.cubicasa5k.create_coco_cc5k --data_root=data/cubicasa5k/ \
    --output=data/coco_cubicasa5k_nowalls_v4/ \
    --disable_wd2line

# Split example has more than 1 floorplan into separate samples
python -m data_preprocess.cubicasa5k.create_coco_cc5k.floorplan_extraction \
    --data_root data/coco_cubicasa5k_nowalls_v4/ \
    --output data/coco_cubicasa5k_nowalls_v4-1_refined/

# Merge individual JSONs into single JSON file per split (train/val/test)
# This must be done after floorplan_extraction.py
python -m data_preprocess.cubicasa5k.combine_json \
    --input data/coco_cubicasa5k_nowalls_v4-1_refined/ \
    --output data/coco_cubicasa5k_nowalls_v4-1_refined/annotations/ \