提交 124f0e82 authored 作者: Glenn Jocher's avatar Glenn Jocher

torchvision nms bug fix

上级 66676eb0
......@@ -8,6 +8,7 @@ import torch
import torch.backends.cudnn as cudnn
import torch.nn as nn
import torch.nn.functional as F
import torchvision
logger = logging.getLogger(__name__)
......@@ -151,7 +152,6 @@ def model_info(model, verbose=False):
def load_classifier(name='resnet101', n=2):
# Loads a pretrained model reshaped to n-class output
import torchvision
model = torchvision.models.__dict__[name](pretrained=True)
# ResNet model properties
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论