search

LymanSong / FP_GNN

A framework for detecting and classifying indoor elements in a floor plan using Graph Neural Network.
MIT License
20 stars 4 forks source link

Prediction on a new image #4

Open jaehyunshinML opened 2 years ago

jaehyunshinML commented 2 years ago

Hi,

Thanks for sharing this code of the amazing research.

I was able to run the test_code.py and tried to run the model with a new floor plan image but got stuck in the middle of it. What I tried:

  1. made a custom folder under dataset folder and put the test image under the fps folder like dataset/test/fps/im/img.png
  2. ran a main.py. I was expecting this generates the processed input file under preprocessed folder. but it was still empty.

Could you give me some direction to test the new floor plan image with your code?

Thank you so much!

Regards,