pytorch 测试 darknet

直接上代码，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 CoordConvclass 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, 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,stri

pytorch 测试 darknet相关推荐

极智AI | pytorch 与 darknet 计算卷积输出 shape 方式对比
欢迎关注我的公众号 [极智视界],回复001获取Google编程规范 O_o >_< o_O O_o ~_~ o_O 本文记录了 pytorch 与 dar ...
pytorch测试代码
pytorch测试代码使用测试pytorch是否正常调用CUDA 1.测试代码代码如下(示例): # Summary: 检测当前Pytorch和设备是否支持CUDA和cudnnimport tor ...
pytorch 测试每一类_DeepFM全方面解析（附pytorch源码）
写在前面最近看了DeepFM这个模型.把我学习的思路和总结放上来给大家和未来的自己做个参考和借鉴.文章主要希望能串起学习DeepFM的各个环节,梳理整个学习思路.以"我"的角度浅 ...
pytorch 测试每一类_2D-UNet脑胶质瘤分割BraTs + Pytorch实现
2D-UNet讲解玖零猴:U-Net+与FCN的区别+医学表现+网络详解+创新zhuanlan.zhihu.com BraTs数据准备数据来源本文用的训练集和验证集均来自BraTs2018的训 ...
pytorch 测试每一类_2D UNet++ VGGBlock脑胶质瘤分割BraTs + Pytorch实现
UNet++讲解玖零猴:UNet++解读 + 它是如何对UNet改进 + 作者的研究态度和方式zhuanlan.zhihu.com BraTs数据准备数据来源本文用的训练集和验证集均来自Bra ...
pytorch测试报错：RuntimeError: cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module
模型在服务器多gpu上训练,测试在自己台式机上进行,只有一块gpu,测试报错: File "/home/fuxueping/sdb/PycharmProjects/face_recognit ...
pytorch测试模型时根据不同列别的概率值得到具体的分类
pytorch 分类任务的教程:https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html 主要使用的是: _,predict ...
pytorch测试模型时显存不够的问题
在训练过程中使用了分块训练,但是在测试的时候使用了整张图片输入到网络中,然后就出现了显存不够的问题经验告诉我不应该不够,但是使用了分块训练,我也不知道是不是真的不够一筹莫展之时,发现没有使用tor ...
pytorch测试，pil 读取图片 “IOError incurred when reading ‘{}‘. Will redo. Don‘t worry. Just chill
#"IOError incurred when reading '{}'. Will redo. Don't worry. Just chill 加入以下问题解决 from PIL imp ...

pytorch 测试 darknet

pytorch 测试 darknet相关推荐

最新文章

热门文章