vgg16 feature extraction pytorch