guqing
6b833d0966
1)init
|
2 days ago | |
|---|---|---|
| .. | ||
| .idea | 2 days ago | |
| __pycache__ | 2 days ago | |
| black-forest-labs | 2 days ago | |
| doc_env | 2 days ago | |
| temp_data | 2 days ago | |
| .inference.lock | 2 days ago | |
| .service_pids | 2 days ago | |
| SG-n6nV8B2oW95.png | 2 days ago | |
| extract_initial_mask.py | 2 days ago | |
| floorplan_env_full.tar | 2 days ago | |
| furniture_detect.onnx | 2 days ago | |
| inference_refine_mask.py | 2 days ago | |
| initial_mask.onnx | 2 days ago | |
| json.md | 2 days ago | |
| pipeline.py | 2 days ago | |
| readme.md | 2 days ago | |
| room_cls.pt | 2 days ago | |
| run_pipeline.sh | 2 days ago | |
| run_pipeline_simple.sh | 2 days ago | |
| vis.py | 2 days ago | |
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