def get_model(self, sublinear=False):
def init_yolo(M):
for m in M.modules():
if isinstance(m, nn.BatchNorm2d):
m.eps = 1e-3
m.momentum = 0.03
if "model" not in self.__dict__:
from yolox.models import YOLOX, YOLOPAFPN, YOLOXHead
in_channels =
# NANO model use depthwise = True, which is main difference.
backbone = YOLOPAFPN(
self.depth, self.width, in_channels=in_channels,
act=self.act, depthwise=True,
)
head = YOLOXHead(
self.num_classes, self.width, in_channels=in_channels,
act=self.act, depthwise=True
)
self.model = YOLOX(backbone, head)
self.model.apply(init_yolo)
self.model.head.initialize_biases(1e-2)
return self.model
谢谢分享哦
感谢分享,正好要用
111111111111111111111111111
厉害厉害厉害厉害厉害厉害
阿达阿瓦达
666666666666666
感谢理工!~!~!~!~!
先用用看再回复
感谢大佬分享
6666666666
11111111111
加油加油加油
谢谢谢谢谢谢谢
111111111111111111
66666666666666666666666666666
66666666666666666
嘟嘟嘟嘟嘟嘟嘟
支持大佬,就是不知道22还能不能用
66666666666666666