huangkuns / wireframe

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Questions about wireframe

haopo2005 opened this issue · comments

Hi,
I'd like to have a quick verification with my own dataset based on your "wireframe".
1.
However, I cant find the any pre-trained models(*.pkl or *.pth files). Can you release them?
2.
During the inference stage, will the output image size of arbitrary input images be unified to 320.*320(default imgdim in opts.py )?
3.
Do you have plans to support the panorama images?
4.
If I have to train from scratch, I'd like to know how long it will take to train on single GPU of P4 processor ?

Best Regards,
Jin

I've trained the model based on their dataset and run the inference stage successfully.
The 8GB GPU memory is not enough for the linepx training, I have to decrease the num of batch size.
As for the result, it provides you with many folders under different thresholds.