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