pytorch 测试 darknet

最新推荐文章于 2024-09-23 00:15:00 发布

AI算法网奇

最新推荐文章于 2024-09-23 00:15:00 发布

阅读量4.2k

点赞数

分类专栏： pytorch知识宝典

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

版权

pytorch知识宝典专栏收录该内容

503 篇文章 228 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍如何在PyTorch平台上使用代码测试Darknet21和Darknet53模型，展示特征维度以及CPU和GPU上的执行时间。尽管Darknet21在嵌入式设备上无法实现实时检测，但提供了模型性能的评估。

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

直接上代码，pytorch平台，测试darknet21，darknet53

本文代码能输出特征维度，cpu、gpu上执行时间

darknet21在嵌入式芯片上还达不到实时检测。

代码改自：

https://github.com/manaai-cn/YoloV4_pytorch/blob/34b95da692df24d09a612d9927b6e3e626ae69e7/models/backbones/darknet.py

import time
import torch
import torch.nn as nn
import math
from collections import OrderedDict

# from nets.coordConv import CoordConv

class BasicBlock(nn.Module):
    def __init__(self, inplanes, planes,dim):
        super(BasicBlock, self).__init__()

        self.conv1 = nn.Conv2d(inplanes, planes[0], kernel_size=1,stride=1, pa