search

yeezhu / SPN.pytorch

PyTorch implementation of "Soft Proposal Networks for Weakly Supervised Object Localization", ICCV 2017.
http://yzhu.work/spn.html
MIT License
211 stars 37 forks source link

Training on new dataset #17

Closed Supersak80 closed 6 years ago

Supersak80 commented 6 years ago

Hi @yeezhu,

Are you able to provide a generic script that allows for training on a custom dataset other than PASCAL VOC, assuming that the images for each class are stored in their respective folders?

Thanks, Sam

yeezhu commented 6 years ago

@Supersak80 See here, a demo dataloader for MITIndoor67.