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

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