StemBlock pytorch实现

柔光的暖阳◎ 2022-10-10 06:12 196阅读 0赞

本文mark一下peleeent和yolov5中的实现：

import time
    from collections import OrderedDict
    import torch.nn.functional as m_func
    import torch
    import cv2
    import torch.nn as nn
    import torch.nn.functional as F
    import numpy as np
    
    
    class conv_bn_relu(nn.Module):
        def __init__(self, nin, nout, kernel_size, stride, padding, bias=False):
            super(conv_bn_relu, self).__init__()
            self.conv = nn.Conv2d(nin, nout, kernel_size=kernel_size, stride=stride, padding=padding, bias=bias)
            self.batch_norm = nn.BatchNorm2d(nout)
            self.relu = nn.ReLU6(True)
            # self.relu = mish.Mish()
    
        def forward(self, x):
            out = self.conv(x)
            out = self.batch_norm(out)
            out = self.relu(out)
    
            return out
    
    class StemBlock(nn.Module):
        def __init__(self):
            super(StemBlock, self).__init__()
    
            self.conv_3x3_first = conv_bn_relu(nin=3, nout=32+32,

发表评论取消回复

表情：

评论列表（有 0 条评论，196人围观）

还没有评论，来说两句吧...

相关阅读

相关 pytorch实现SDNE代码

好的，下面是用 PyTorch 实现 SDNE (Structure Deep Network Embedding) 的代码。首先，我们需要定义一个 SDNE 模型类。

淡淡的烟草味﹌/ 2024年03月26日 09:14/ 0 赞/ 60 阅读

相关 pytorch实现CBAM

CBAM (Convolutional Block Attention Module) 是一种用于计算卷积层中特征图的注意力权重的方法。在 PyTorch 中实现 CBAM 需

﹏ヽ暗。殇╰゛Y/ 2024年03月26日 08:37/ 0 赞/ 66 阅读

相关 PyTorch的简单实现

[原文链接][Link 1] 1.必要的 PyTorch 背景 1. PyTorch 是一个建立在 Torch 库之上的 Python 包，旨在加速深度学习应用。 2

阳光穿透心脏的1/2处/ 2023年10月18日 15:51/ 0 赞/ 112 阅读

相关 pytorch主流backbone实现

计算机视觉主流backbone收集源码 [https://github.com/oaifaye/backbone-pytorch][https_github.com_oai

灰太狼/ 2023年10月02日 10:35/ 0 赞/ 11 阅读

相关 PyTorch 04：用PyTorch实现Fashion-MNIST

好了，我们来实战演练。你将使用 [Fashion-MNIST 数据集][Fashion-MNIST]。对于神经网络来说，MNIST 的规模很小，可以轻松地达到 97% 的准确率

傷城~/ 2023年03月13日 15:29/ 0 赞/ 16 阅读

相关 Pytorch实现CSPDarknet53

import torch import torch.nn.functional as F import torch.nn as nn impo

本是古典何须时尚/ 2023年01月13日 10:58/ 0 赞/ 218 阅读

相关 pytorch实现unet

unet是非常经典的图像分割的网络，因网络结构形似字母U而著称实现起来不是很复杂，代码如下： -- coding: utf-8 -- import to

我会带着你远行/ 2022年10月28日 12:23/ 0 赞/ 292 阅读

相关 StemBlock pytorch实现

本文mark一下peleeent和yolov5中的实现： import time from collections import OrderedDict

柔光的暖阳◎/ 2022年10月10日 06:12/ 0 赞/ 197 阅读

相关 pytorch 实现LSTM

\\ Pytorch基础知识点整理 \\ 梯度下降： __coding:utf-8__ from math import pi impo

客官°小女子只卖身不卖艺/ 2022年03月28日 09:44/ 0 赞/ 280 阅读

相关 PyTorch 实现 Mask-RCNN

文章目录 \[[隐藏][Link 1]\] [1 什么是 Mask-RCNN][1 _ Mask-RCNN] [2 PyTorch 实现 Mask-RCNN][2

柔情只为你懂/ 2022年02月25日 16:10/ 0 赞/ 341 阅读