/VGG_feature_extractor

VGG19 feature extractor using PyTorch framework

Primary LanguagePython

Example code for extracting VGG features by using PyTorch framework

Configuration

  • image_path : the path of image want to extract VGG feature

  • feature_layer : the layer of VGG network want to extract the feature (e.g,. relu2_2, conv3_2, ... )

Comments

If you have any questions or comments on my codes, please email to me. son1113@snu.ac.kr