yolov3 测试

darknet53 在我1060上gpu 352*352 需要21ms，cpu需要900ms，但是参数比较多，占用内存比较大？
 

总时间需要30多ms

      import datetime
      import torch
      import torch.nn as nn
      import math
      from collections import OrderedDict
      __all__ = ['darknet21', 'darknet53']
      class BasicBlock(nn.Module):
      def __init__(self, inplanes, planes):
       super(BasicBlock, self).__init__()
       self.conv1 = nn.Conv2d(inplanes, planes[0], kernel_size=1,
       stride=1, padding=0, bias=False)
       self.bn1 = nn.BatchNorm2d(planes[0])
       self.relu1 = nn.LeakyReLU(0.1)
       self.conv2 = nn.Conv2d(planes[0], planes[1], kernel_size=3,
       stride=1, padding=1, bias=False)
       self.bn2 = nn.BatchNorm2d(planes[1])
       self.relu2 = nn.LeakyReLU(0.1)
      def forward(se
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/87910061