OCtaveResNet 测试

最新推荐文章于 2021-08-25 10:13:34 发布

AI算法网奇

最新推荐文章于 2021-08-25 10:13:34 发布

阅读量2.6k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：深度学习宝典

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

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

1041 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客介绍了OctaveResNet在256*256分辨率下，相较于resnet50速度从17ms增加到52ms的测试结果。同时提到了resnet34在相同条件下也需要33ms。提供了相关的代码仓库链接。

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

代码地址：

https://github.com/lxtGH/OctaveConv_pytorch

256*256 resnet50的速度原来是17ms，新的52ms

下面这个需要33ms

if __name__ == '__main__':
    model = DataSetAwareResnet50().cuda()
    import time

    for i in range(10):
        img = torch.Tensor(1, 3, 256, 256).cuda()
        with torch.no_grad():
            torch.cuda.synchronize()
            start = time.time()
            out = model(img)
            torch.cuda.synchronize()
            print('time.{}'.format(time.time() - start))

resnet34本机需要33ms

代码：

了解本专栏