search

hao-liu-china / SiameseFC-pytorch

52 stars 14 forks source link

Advice to speed up #1

Open HelloRicky123 opened 6 years ago

HelloRicky123 commented 6 years ago

I think you can put as many as possible part in getitem method of dataset.py to init.py. And use torch.grid_sample may speed up the padding and crop process.

hao-liu-china commented 6 years ago

@HelloRicky123 Thank you for your advice, I will try it as soon as possible.