├── .gitignore ├── 00_preprocess ├── generate_masks.py ├── mask_fixer.py └── utils.py ├── 01_instance_seg ├── bin_mean_shift.py ├── data_loader.py ├── data_loader_test.py ├── metric_loss.py ├── models │ ├── github_model.py │ ├── unet │ │ ├── __init__.py │ │ ├── nano.save │ │ ├── unet_model.py │ │ └── unet_parts.py │ └── unet_coord │ │ ├── __init__.py │ │ ├── nano.save │ │ ├── unet_model.py │ │ └── unet_parts.py ├── predict_new.py ├── train.py ├── utils │ ├── __init__.py │ ├── config.py │ ├── config.yaml │ └── misc.py └── vis_cat.py ├── 02_type_class ├── data_gen.py ├── data_loader.py ├── evaluate.py ├── example.yaml ├── paths.py ├── process_boundary.py ├── train.py └── utils.py ├── 03_frame_detect ├── combined_loader.py ├── data_utils.py ├── example.yaml ├── find_corr.py ├── paths.py ├── sem_rings.py ├── train_cnn.py └── utils.py ├── 04_frame_correct ├── data_loader.py ├── data_utils.py ├── example.yaml ├── find_corr.py ├── graph_utils.py ├── models_img.py ├── refine_utils.py ├── refinery.py ├── sem_rings.py ├── train.py └── utils.py ├── 05_simplification ├── actions.py ├── parallel.py ├── timer.py └── utils.py ├── 06_eval ├── door_eval.py ├── eval_mrcnn.py ├── eval_r2v.py ├── find_corr.py ├── main.py ├── metric.py ├── mrcnn_door_eval.py ├── new_utils.py ├── qual_vis.py ├── r2v_door_eval.py ├── sem_rings.py ├── stats.py └── vectorize.py ├── ANNOTATE.md ├── README.md └── resources ├── add_attribute.png └── annotate.gif /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/.gitignore -------------------------------------------------------------------------------- /00_preprocess/generate_masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/00_preprocess/generate_masks.py -------------------------------------------------------------------------------- /00_preprocess/mask_fixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/00_preprocess/mask_fixer.py -------------------------------------------------------------------------------- /00_preprocess/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/00_preprocess/utils.py -------------------------------------------------------------------------------- /01_instance_seg/bin_mean_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/bin_mean_shift.py -------------------------------------------------------------------------------- /01_instance_seg/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/data_loader.py -------------------------------------------------------------------------------- /01_instance_seg/data_loader_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/data_loader_test.py -------------------------------------------------------------------------------- /01_instance_seg/metric_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/metric_loss.py -------------------------------------------------------------------------------- /01_instance_seg/models/github_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/models/github_model.py -------------------------------------------------------------------------------- /01_instance_seg/models/unet/__init__.py: -------------------------------------------------------------------------------- 1 | from .unet_model import UNet 2 | -------------------------------------------------------------------------------- /01_instance_seg/models/unet/nano.save: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /01_instance_seg/models/unet/unet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/models/unet/unet_model.py -------------------------------------------------------------------------------- /01_instance_seg/models/unet/unet_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/models/unet/unet_parts.py -------------------------------------------------------------------------------- /01_instance_seg/models/unet_coord/__init__.py: -------------------------------------------------------------------------------- 1 | from .unet_model import UNet 2 | -------------------------------------------------------------------------------- /01_instance_seg/models/unet_coord/nano.save: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /01_instance_seg/models/unet_coord/unet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/models/unet_coord/unet_model.py -------------------------------------------------------------------------------- /01_instance_seg/models/unet_coord/unet_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/models/unet_coord/unet_parts.py -------------------------------------------------------------------------------- /01_instance_seg/predict_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/predict_new.py -------------------------------------------------------------------------------- /01_instance_seg/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/train.py -------------------------------------------------------------------------------- /01_instance_seg/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /01_instance_seg/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/utils/config.py -------------------------------------------------------------------------------- /01_instance_seg/utils/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/utils/config.yaml -------------------------------------------------------------------------------- /01_instance_seg/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/utils/misc.py -------------------------------------------------------------------------------- /01_instance_seg/vis_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/01_instance_seg/vis_cat.py -------------------------------------------------------------------------------- /02_type_class/data_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/02_type_class/data_gen.py -------------------------------------------------------------------------------- /02_type_class/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/02_type_class/data_loader.py -------------------------------------------------------------------------------- /02_type_class/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/02_type_class/evaluate.py -------------------------------------------------------------------------------- /02_type_class/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/02_type_class/example.yaml -------------------------------------------------------------------------------- /02_type_class/paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/02_type_class/paths.py -------------------------------------------------------------------------------- /02_type_class/process_boundary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/02_type_class/process_boundary.py -------------------------------------------------------------------------------- /02_type_class/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/02_type_class/train.py -------------------------------------------------------------------------------- /02_type_class/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/02_type_class/utils.py -------------------------------------------------------------------------------- /03_frame_detect/combined_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/03_frame_detect/combined_loader.py -------------------------------------------------------------------------------- /03_frame_detect/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/03_frame_detect/data_utils.py -------------------------------------------------------------------------------- /03_frame_detect/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/03_frame_detect/example.yaml -------------------------------------------------------------------------------- /03_frame_detect/find_corr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/03_frame_detect/find_corr.py -------------------------------------------------------------------------------- /03_frame_detect/paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/03_frame_detect/paths.py -------------------------------------------------------------------------------- /03_frame_detect/sem_rings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/03_frame_detect/sem_rings.py -------------------------------------------------------------------------------- /03_frame_detect/train_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/03_frame_detect/train_cnn.py -------------------------------------------------------------------------------- /03_frame_detect/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/03_frame_detect/utils.py -------------------------------------------------------------------------------- /04_frame_correct/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/data_loader.py -------------------------------------------------------------------------------- /04_frame_correct/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/data_utils.py -------------------------------------------------------------------------------- /04_frame_correct/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/example.yaml -------------------------------------------------------------------------------- /04_frame_correct/find_corr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/find_corr.py -------------------------------------------------------------------------------- /04_frame_correct/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/graph_utils.py -------------------------------------------------------------------------------- /04_frame_correct/models_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/models_img.py -------------------------------------------------------------------------------- /04_frame_correct/refine_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/refine_utils.py -------------------------------------------------------------------------------- /04_frame_correct/refinery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/refinery.py -------------------------------------------------------------------------------- /04_frame_correct/sem_rings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/sem_rings.py -------------------------------------------------------------------------------- /04_frame_correct/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/train.py -------------------------------------------------------------------------------- /04_frame_correct/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/04_frame_correct/utils.py -------------------------------------------------------------------------------- /05_simplification/actions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/05_simplification/actions.py -------------------------------------------------------------------------------- /05_simplification/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/05_simplification/parallel.py -------------------------------------------------------------------------------- /05_simplification/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/05_simplification/timer.py -------------------------------------------------------------------------------- /05_simplification/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/05_simplification/utils.py -------------------------------------------------------------------------------- /06_eval/door_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/door_eval.py -------------------------------------------------------------------------------- /06_eval/eval_mrcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/eval_mrcnn.py -------------------------------------------------------------------------------- /06_eval/eval_r2v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/eval_r2v.py -------------------------------------------------------------------------------- /06_eval/find_corr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/find_corr.py -------------------------------------------------------------------------------- /06_eval/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/main.py -------------------------------------------------------------------------------- /06_eval/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/metric.py -------------------------------------------------------------------------------- /06_eval/mrcnn_door_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/mrcnn_door_eval.py -------------------------------------------------------------------------------- /06_eval/new_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/new_utils.py -------------------------------------------------------------------------------- /06_eval/qual_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/qual_vis.py -------------------------------------------------------------------------------- /06_eval/r2v_door_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/r2v_door_eval.py -------------------------------------------------------------------------------- /06_eval/sem_rings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/sem_rings.py -------------------------------------------------------------------------------- /06_eval/stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/stats.py -------------------------------------------------------------------------------- /06_eval/vectorize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/06_eval/vectorize.py -------------------------------------------------------------------------------- /ANNOTATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/ANNOTATE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/README.md -------------------------------------------------------------------------------- /resources/add_attribute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/resources/add_attribute.png -------------------------------------------------------------------------------- /resources/annotate.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weiliansong/blueprint-vectorizer/HEAD/resources/annotate.gif --------------------------------------------------------------------------------