dongheehand / VGG_feature_extractor

VGG19 feature extractor using PyTorch framework

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. [e-mail]