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

readme.md

需要安装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