efficientdet.pytorch icon indicating copy to clipboard operation
efficientdet.pytorch copied to clipboard

efficient net as feature extractor

Open kumarmanas opened this issue 4 years ago • 0 comments
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)

kumarmanas avatar Jun 01 '21 23:06 kumarmanas