search

huangkuns / wireframe

217 stars 45 forks source link

Questions about wireframe #3

Closed haopo2005 closed 6 years ago

haopo2005 commented 6 years ago

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

haopo2005 commented 6 years ago

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.