guqing/hongchen

guqing 6b833d0966 1)init		2 dagar sedan
..
.idea	6b833d0966 1)init	2 dagar sedan
__pycache__	6b833d0966 1)init	2 dagar sedan
black-forest-labs	6b833d0966 1)init	2 dagar sedan
doc_env	6b833d0966 1)init	2 dagar sedan
temp_data	6b833d0966 1)init	2 dagar sedan
.inference.lock	6b833d0966 1)init	2 dagar sedan
.service_pids	6b833d0966 1)init	2 dagar sedan
SG-n6nV8B2oW95.png	6b833d0966 1)init	2 dagar sedan
extract_initial_mask.py	6b833d0966 1)init	2 dagar sedan
floorplan_env_full.tar	6b833d0966 1)init	2 dagar sedan
furniture_detect.onnx	6b833d0966 1)init	2 dagar sedan
inference_refine_mask.py	6b833d0966 1)init	2 dagar sedan
initial_mask.onnx	6b833d0966 1)init	2 dagar sedan
json.md	6b833d0966 1)init	2 dagar sedan
pipeline.py	6b833d0966 1)init	2 dagar sedan
readme.md	6b833d0966 1)init	2 dagar sedan
room_cls.pt	6b833d0966 1)init	2 dagar sedan
run_pipeline.sh	6b833d0966 1)init	2 dagar sedan
run_pipeline_simple.sh	6b833d0966 1)init	2 dagar sedan
vis.py	6b833d0966 1)init	2 dagar sedan

需要安装NVIDIA 显卡，NVIDIA 驱动，NVIDIA Container Toolkit

docker load -i floorplan_env_full.tar 将图片（比如pic_name.png）复制到/temp_data下 eg:运行./run_pipeline_simple.sh pic_name.png 保存的json信息位于/temp_data/pic_name/pic_name.json

json的格式和字段信息见json.md

readme.md