Skip to content

Instantly share code, notes, and snippets.

@yearofthewhopper

yearofthewhopper/Feature.py

Created February 22, 2020 18:27
Show Gist options
  • Save yearofthewhopper/7d15053844e6267fcfa53871bc50c861 to your computer and use it in GitHub Desktop.
Save yearofthewhopper/7d15053844e6267fcfa53871bc50c861 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Feature.py
class FeatureX(nn.Module):
def __init__(self):
super(FeatureX, self).__init__()
vgg19_model = models.vgg19(pretrained=True)
self.feature_extractor = nn.Sequential(*list(vgg19_model.features.children())[:18])
def forward(self, img):
return self.feature_extractor(img)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment