search

DavidGillsjo / polygon-HGT

Polygon Detection for Room Layout Estimation using Heterogenous Graphs and Wireframes
MIT License
25 stars 4 forks source link

problem with annotation and inference scripts #5

Closed LucBourrat1 closed 11 months ago

LucBourrat1 commented 1 year ago

I have a problem when trying to annotate data using the script structured3Dplanes_simple.py, the following modules are not found:

Where are these modules?

Also I tried running the inference script on my own images using these commands: cd scripts python3 test.py \ --config-file ../config-files/Pred-simple-plane-S3D-GNN.yaml \ --img-folder <my-image-folder> \ CHECKPOINT ../data/model_polygon_hgt_joint.pth

But it does not work because in yaml configuration file some path are not found.

Thanks for your help

DavidGillsjo commented 11 months ago

Sorry about that! I've uploaded the missing files. Please reopen the issue if it doesn't work.