DarknetTiny模型结构

darknet 是yolov3 里用的模型，如下是是其模型结构

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon Jun  7 13:40:34 2021@author: ledi
"""import numpy as np
import tensorflow as tf
from tensorflow.keras import Model
from tensorflow.keras.layers import (Add,Concatenate,Conv2D,Input,Lambda,LeakyReLU,MaxPool2D,UpSampling2D,ZeroPadding2D,BatchNormalization,
)
from tensorflow.keras.regularizers import l2
from tensorflow.keras.losses import (binary_crossentropy,sparse_categorical_crossentropy
)def DarknetConv(x, filters, size, strides=1, batch_norm=True):if strides == 1:padding = 'same'else:x = ZeroPadding2D(((1, 0), (1, 0)))(x)  # top left half-paddingpadding = 'valid'x = Conv2D(filters=filters, kernel_size=size,strides=strides, padding=padding,use_bias=not batch_norm, kernel_regularizer=l2(0.0005))(x)if batch_norm:x = BatchNormalization()(x)x = LeakyReLU(alpha=0.1)(x)return xdef DarknetResidual(x, filters):prev = xx = DarknetConv(x, filters // 2, 1)x = DarknetConv(x, filters, 3)x = Add()([prev, x])return xdef DarknetBlock(x, filters, blocks):x = DarknetConv(x, filters, 3, strides=2)for _ in range(blocks):x = DarknetResidual(x, filters)return xdef Darknet(name=None):x = inputs = Input([None, None, 3])x = DarknetConv(x, 32, 3)x = DarknetBlock(x, 64, 1)x = DarknetBlock(x, 128, 2)  # skip connectionx = x_36 = DarknetBlock(x, 256, 8)  # skip connectionx = x_61 = DarknetBlock(x, 512, 8)x = DarknetBlock(x, 1024, 4)return tf.keras.Model(inputs, (x_36, x_61, x), name=name)def DarknetTiny(name=None):x = inputs = Input([None, None, 3])x = DarknetConv(x, 16, 3)x = MaxPool2D(2, 2, 'same')(x)x = DarknetConv(x, 32, 3)x = MaxPool2D(2, 2, 'same')(x)x = DarknetConv(x, 64, 3)x = MaxPool2D(2, 2, 'same')(x)x = DarknetConv(x, 128, 3)x = MaxPool2D(2, 2, 'same')(x)x = x_8 = DarknetConv(x, 256, 3)  # skip connectionx = MaxPool2D(2, 2, 'same')(x)x = DarknetConv(x, 512, 3)x = MaxPool2D(2, 1, 'same')(x)x = DarknetConv(x, 1024, 3)return tf.keras.Model(inputs, (x_8, x), name=name)model=DarknetTiny(name=None)from keras.utils import plot_model
plot_model(model, to_file='darknet.png')

DarknetTiny模型结构相关推荐

理论与实践中的CNN模型结构，如何引领深度学习热潮
摘要: 深度学习是指多层神经网络上运用各种机器学习算法解决图像,文本等各种问题的算法集合.卷积神经网络(CNN)是深度学习框架中的一个重要算法,本文介绍了CNN主流模型结构的演进过程,从一切的开始Le ...
【深度学习入门到精通系列】模型结构可视化神器Netron（连.pth都可以~！)
文章目录 1 Netron 2 说明 1 Netron 目前的Netron支持主流各种框架的模型结构可视化工作,我直接给出gayhub链接: https://github.com/lutzroeder ...
卷积神经网络(CNN)模型结构
在前面我们讲述了DNN的模型与前向反向传播算法.而在DNN大类中,卷积神经网络(Convolutional Neural Networks,以下简称CNN)是最为成功的DNN特例之一.CNN广泛的应用 ...
【NLP-词向量】从模型结构到损失函数详解word2vec
上周我们讲到,在进行NNLM训练时,能够得到副产品,词向量.本文介绍一种专门用于词向量制备的方法:word2vec,利用它能够高效的训练出词向量. 作者&编辑 | 小Dream哥 1 word ...
【GAN模型结构】从最简单的全卷积GAN一起开始玩转GAN
大家好,今天开设新专栏<GAN模型结构>.在这个专栏中,我们会讲述各种各样的GAN的网络结构及其在各类任务中的应用. 作者&编辑 | 言有三三年前GAN刚出来的时候就开始关注GA ...
脑洞大开的机器视觉多领域学习模型结构 | CVPR 2018论文解读
在碎片化阅读充斥眼球的时代,越来越少的人会去关注每篇论文背后的探索和思考. 在这个栏目里,你会快速 get 每篇精选论文的亮点和痛点,时刻紧跟 AI 前沿成果. 点击本文底部的「阅读原文」即刻加入社区 ...
echart label 不随值变化位置_炼丹神器来了！模型结构、训练日志、特征提取都能可视化，调参不慌了！...
点击上方"CVer",选择加"星标"置顶重磅干货,第一时间送达开发者在炼丹的过程中会不会遇到这些问题呢?比如说深度学习模型训练过程中盯 Log 盯得头晕眼花 ...
可视化caffe模型结构及在线可视化
在线可视化caffe模型结构 http://ethereon.github.io/netscope/#/editor 假设Caffe的目录是$(CAFFE_ROOT) 1.编译caffe的python ...
动手学CV-目标检测入门教程4：模型结构
3.4 模型结构本文来自开源组织 DataWhale

DarknetTiny模型结构

darknet 是yolov3 里用的模型，如下是是其模型结构

DarknetTiny模型结构相关推荐

最新文章

热门文章