Imported from https://github.com/fulviomascara/pytorch-cv