How to predict results for a single image?

[thevoxium]

Hi,

I was trying to get the results on a single image but it seems the predict_all.py uses the complete dataset for the evaluation. The repo doesn't have a prediction function for a single image like PlaneAE. Can you tell me are you planning to work on that or suggest ways so I can implement it myself.

Thanks beforehand.

[yi-ming-qian]

Hi, I guess you can easily write a simple script by following predict_all.py to do such a thing.
Our method relies on PlaneRCNN's result, so you first need to run PlaneRNN/PlaneAE for a single image to get their results.