Thursday 17 December 2020 at 12:43

Multi-Scale Training with PyTorch Image Folder

By Eric Antoine Scuccimarra

I've had good luck with multi-scale training for image detection so I wanted to try it for classification of images that were of different sizes with objects at differing scales. I found some base code here , but this is based on PyTorch datasets, not on ImageFolders. I wrote some code to extend it to ImageFolders, which is in the below gist :

Labels: machine_learning, pytorch


Login or Register to leave a comment..