vovnet 测试

AI算法网奇

于 2020-02-28 11:30:21 发布

阅读量3.2k

点赞数

分类专栏：深度学习宝典

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/104552883

版权

深度学习宝典专栏收录该内容

1032 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

博主分享了关于Vovnet的个人参数修改经验，对比原版模型，调整后的配置在最大batch为4的情况下，运行时间从130ms有所变化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

vovnet39， 1070 640*640 batch 1 15ms

自己改了参数：

import time

import torch
import torch.nn as nn
import torch.nn.functional as F
from collections import OrderedDict

__all__ = ['VoVNet', 'vovnet27_slim', 'vovnet39', 'vovnet57']



def conv3x3(in_channels, out_channels, module_name, postfix,
            stride=1, groups=1, kernel_size=3, padding=1):
    """3x3 convolution with padding"""
    return [
        ('{}_{}/conv'.format(module_name, postfix),
         nn.Conv2d(in_channels, out_channels,
                   kernel_size=kernel_size,
                   stride=stride,
                   paddi