search

lood339 / SCCvSD

Sports Camera Calibration via Synthesic Data
BSD 2-Clause "Simplified" License
71 stars 19 forks source link

Memory Usage Requirements? #14

Closed nlin5 closed 3 years ago

nlin5 commented 3 years ago

Hi Dr. Chen,

Thank you for your contributions by providing this repository along with your paper. I am able to generate the train_data_10k.mat file containing pivot images, positive images, and cameras. However, when inputting that .mat file into network_train.py and running the script, I appear to run into memory limitations and the script is Killed. I was just wondering what the specs of your machine is when you trained the Siamese network?

Thanks

lood339 commented 3 years ago

Hi nlin5, Thank you for interested in our work. It is quite strange to me that the program runs into memory limitation. The network and the input image resolution are quite small. Please double check that the input image size is 320x180. If your GPU has limited memory, please use small batch size to train the siemese network. Hope that helps.