YOLOv5 (6.0/6.1) 简要摘要

导入时间
导入火炬
导入 pillage.nn 为 nn


类 SPP(nn.Module):
\\ nit ( 自己) :
超级 (. ) init  ()
self.maxpool1=nn.MaxPool2d (5, 1, padding=2) 互联网档案馆的存檔,存档日期2013-12-2.
self.maxpool2=nn.MaxPool2d(9, 1, padding=4) 互联网档案馆的存檔,存档日期2013-12-02.
self.maxpool3=nn.MaxPool2d(13,1,padding=6) 互联网档案馆的存檔,存档日期2013-12-20.

向前( 自, x) :
o1 = 自.maxpool1(x)
o2 = 自.maxpool2(x)
o3 = 自.maxpool3(x)
返回火炬. cat([x, o1, o2, o3], dim=1)


类 SPPF(nn.Module):
\\ nit ( 自己) :
超级 (. ) init  ()
self.maxpool=nn.MaxPool2d (5, 1, padding=2) 互联网档案馆的存檔,存档日期2013-12-2.

向前( 自, x) :
o1 = 自定义(x)
o2 = 自我.maxpool(o1)
o3 = 自我.maxpool(o2)
返回火炬. cat([x, o1, o2, o3], dim=1)


daf main () (英语:
输入( tensor = hapt.rand (8, 32, 16, 16)
spp = SPP( )
sppf = SPPF ()
输出1 = spp(输入数)
输出2 = sppf(输入值)

打印(torch.equal(输出1,输出2))

t  start = 时间. time ()
 在范围(100):
spp(输入频率)
打印( f) “ spp 时间 : {time.time () - t  start}' )

t  start = 时间. time ()
 在范围(100):
sppf(输入频率)
打印( f) sppf 时间 : {time.time () - t  start}' )


如果... 名字... \"... main...":
主要( )

对
spp时间: 0.5373051166534424
sppf 时间: 0.20780706405639648