vovnet 测试

最新推荐文章于 2024-08-29 16:26:24 发布

AI算法网奇

最新推荐文章于 2024-08-29 16:26:24 发布

阅读量3.1k

点赞数

分类专栏：深度学习宝典

本文链接：https://blog.csdn.net/jacke121/article/details/104552883

版权

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

998 篇文章 717 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

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

摘要由CSDN通过智能技术生成

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

了解本专栏