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

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