efficientdet.pytorch
efficientdet.pytorch copied to clipboard
efficient net as feature extractor
trafficstars
Thanks for your torch implementation. I am interested in getting features of image same as we do in Resnet. Is it possible to do via your implementation? If yes, can you please let me know how to use that. Like in resnet I can do like this way-
from model.res50 import ResNet
self.encoder = ResNet()
self.fc = nn.Linear(hidden_dim, num_classes)