深度学习(七十)darknet 实现编写mobilenet源码
一、添加一个新的网络层
(1)parse.c文件中函数string_to_layer_type,添加网络层类型解析:
if (strcmp(type, "[depthwise_convolutional]") == 0) return DEPTHWISE_CONVOLUTIONAL;
(2)darknet.h文件中枚举类型LAYER_TYPE,添加网络层枚举类型:
DEPTHWISE_CONVOLUTIONAL;
(3)parse.c文件中函数parse_network_cfg添加网络层解析后进行构建:
LAYER_TYPE lt = string_to_layer_type(s->type);
if (lt == DEPTHWISE_CONVOLUTIONAL) {
l = parse_depthwise_convolutional(options, params);//自己编写的函数,主要调用了make layer
}
else if(lt == CONVOLUTIONAL){
l = parse_convolutional(options, params);
(4)parse.c 添加参数读取网络层加载文件
A、添加网络层读取参数函数
void load_weights_upto(network *net, char *filename, int start, int cutoff) load depthwise weights;
B、把参数读取到内存上:
void load_depthwise_convolutional_weights(layer l, FILE *fp);
C、以及修改depthwise_convolutional_kenel.cu把读取后的cpu参数拷贝到显存上:
void push_depthwise_convolutional_layer(depthwise_convolutional_layer layer);
(5)parse.c添加参数保存功能:
void save_weights_upto(network net, char *filename, int cutoff): void save_depthwise_convolutional_weights(layer l, FILE *fp); void pull_depthwise_convolutional_layer(depthwise_convolutional_layer layer);
(6)添加network.c中网络层:
int resize_network(network *net, int w, int h)的resize:
if (l.type==DEPTHWISE_CONVOLUTIONAL)
{
resize_depthwise_convolutional_layer(&l, w, h);
}
(7)另外在多卡异步训练的时候,network_kernels.cu的好几个函数也要添加depth_convolutional参数相关的更新设置。
总结为一句话:直接搜索项目中调用:CONVOLUTIONAL的关键子
有调用到卷积层枚举类型的地方,可分离卷积层也要添加相对应的功能。
(8)darknet使用须知:darknet的网络配置文件中的学习率、batch并不是我们平时所说的学习率、batch_size。网络更新所用的学习率为:learning_rate/batch_size,所以学习率不能太小,比如如果学习率设置为0.01,batch=128,那么实际计算的学习率就是0.000078,非常小的一个数值,基本上就是更新不了
二、编写网络层代码:depthwise_convolutional_kernels.cu、depthwise_convolutional_layer.c、depthwise_convolutional_layer.h
三、编写mobilenet网络结构文件:
[net] batch=32 subdivisions=1 height=224 width=224 channels=3 momentum=0.9 decay=0.000 max_crop=320 learning_rate=0.1 policy=poly power=3 max_batches=1600000 #conv1 [convolutional] batch_normalize=1 filters=32 size=3 stride=2 pad=1 activation=relu #conv2_1/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=1 pad=1 activation=relu #conv2_1/sep [convolutional] batch_normalize=1 filters=64 size=1 stride=1 pad=0 activation=relu #conv2_2/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=2 pad=1 activation=relu #conv2_2/sep [convolutional] batch_normalize=1 filters=128 size=1 stride=1 pad=0 activation=relu #conv3_1/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=1 pad=1 activation=relu #conv3_1/sep [convolutional] batch_normalize=1 filters=128 size=1 stride=1 pad=0 activation=relu #conv3_2/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=2 pad=1 activation=relu #conv3_2/sep [convolutional] batch_normalize=1 filters=256 size=1 stride=1 pad=0 activation=relu #conv4_1/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=1 pad=1 activation=relu #conv4_1/sep [convolutional] batch_normalize=1 filters=256 size=1 stride=1 pad=0 activation=relu #conv4_2/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=2 pad=1 activation=relu #conv4_2/sep [convolutional] batch_normalize=1 filters=512 size=1 stride=1 pad=0 activation=relu #conv5_1/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=1 pad=1 activation=relu #conv5_1/sep [convolutional] batch_normalize=1 filters=512 size=1 stride=1 pad=0 activation=relu #conv5_2/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=1 pad=1 activation=relu #conv5_2/sep [convolutional] batch_normalize=1 filters=512 size=1 stride=1 pad=0 activation=relu #conv5_3/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=1 pad=1 activation=relu #conv5_3/sep [convolutional] batch_normalize=1 filters=512 size=1 stride=1 pad=0 activation=relu #conv5_4/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=1 pad=1 activation=relu #conv5_4/sep [convolutional] batch_normalize=1 filters=512 size=1 stride=1 pad=0 activation=relu #conv5_5/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=1 pad=1 activation=relu #conv5_5/sep [convolutional] batch_normalize=1 filters=512 size=1 stride=1 pad=0 activation=relu #conv5_6/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=2 pad=1 activation=relu #conv5_6/sep [convolutional] batch_normalize=1 filters=1024 size=1 stride=1 pad=0 activation=relu #conv6/dw [depthwise_convolutional] batch_normalize=1 size=3 stride=1 pad=1 activation=relu #conv6/sep [convolutional] batch_normalize=1 filters=1024 size=1 stride=1 pad=0 activation=relu #pool6 [avgpool] #fc7 [convolutional] filters=1000 size=1 stride=1 pad=0 activation=leaky [softmax] groups=1 [cost]
四、imagenet训练实验
1、训练一天后,经过两轮多的epoch后,精度:
2、训练两天后,迭代第二天结果:
3\又训练了一天多:
深度学习(七十)darknet 实现编写mobilenet源码的更多相关文章
- Tensorflow深度学习之十二:基础图像处理之二
Tensorflow深度学习之十二:基础图像处理之二 from:https://blog.csdn.net/davincil/article/details/76598474 首先放出原始图像: ...
- 并发编程学习笔记(9)----AQS的共享模式源码分析及CountDownLatch使用及原理
1. AQS共享模式 前面已经说过了AQS的原理及独享模式的源码分析,今天就来学习共享模式下的AQS的几个接口的源码. 首先还是从顶级接口acquireShared()方法入手: public fin ...
- 并发编程学习笔记(8)----ThreadLocal的使用及源码分析
1. ThreadLocal的理解 ThreadLocal,顾名思义,就是线程的本地变量,ThreadLocal会为每个线程创建一个本地变量副本,使得使用ThreadLocal管理的变量在多线程的环境 ...
- 【转载】深度解读 java 线程池设计思想及源码实现
总览 开篇来一些废话.下图是 java 线程池几个相关类的继承结构: 先简单说说这个继承结构,Executor 位于最顶层,也是最简单的,就一个 execute(Runnable runnable) ...
- Java并发指南12:深度解读 java 线程池设计思想及源码实现
深度解读 java 线程池设计思想及源码实现 转自 https://javadoop.com/2017/09/05/java-thread-pool/hmsr=toutiao.io&utm_ ...
- Java并发包源码学习系列:JDK1.8的ConcurrentHashMap源码解析
目录 为什么要使用ConcurrentHashMap? ConcurrentHashMap的结构特点 Java8之前 Java8之后 基本常量 重要成员变量 构造方法 tableSizeFor put ...
- Deep learning深度学习的十大开源框架
Google开源了TensorFlow(GitHub),此举在深度学习领域影响巨大,因为Google在人工智能领域的研发成绩斐然,有着雄厚的人才储备,而且Google自己的Gmail和搜索引擎都在使用 ...
- JavaScript学习总结(十四)——JavaScript编写类的扩展方法
在JavaScript中可以使用类的prototype属性来扩展类的属性和方法,在实际开发当中,当JavaScript内置的那些类所提供的动态 ...
- 深度学习(十) GoogleNet
GoogLeNet Incepetion V1 这是GoogLeNet的最早版本,出现在2014年的<Going deeper with convolutions>.之所以名为“GoogL ...
随机推荐
- 探讨"点"语法的奥秘
点语法 一直以来,都不理解什么是点语法,都说是相当于链接或是路径.也许我浏览的信息量少吧,看过好几篇有关的博文,没什么记载,本篇只是初步见解分析. 在javascript里,属性和方法都使用“点”语法 ...
- nsis源码 nsisdialogdesigner
; 安装程序初始定义常量!define PRODUCT_NAME "nsisdialogdesigner"!define PRODUCT_VERSION "1.1.3&q ...
- HDFS并行复制Distcp
1)Distcp(分布式拷贝)是用于大规模集群内部和集群之间拷贝的工具. 2)distcp命令是以MR作业(没有R任务)的形式实现的,把文件和目录的列表作为M任务的输入.每一个文件是由一个M任务来拷贝 ...
- js去除前后空格
<script language="javascript"> String.prototype.trim=function(){ return this.rep ...
- c#的逆向工程-IL指令集
一些 IL 语言解释: 跳转指令集合 Public field Static Beq 如果两个值相等,则将控制转移到目标指令. Public field Static Beq ...
- P4Lang Repository: Switch
Github Switch Introduction Structure: +-----+ +-----+ +-----+ +-----+ |App a| |App j| |App n| |App z ...
- 03_MySQL DQL_排序查询
#进阶3:排序查询/*语法: select 查询列表 from 表名 [where 筛选条件] order by 排序列表 [asc|desc] 特点: 1.asc升序,desc降序, 如果都不写,默 ...
- linux安装数据库删除
https://blog.csdn.net/qq_40550973/article/details/80721014 卸载mysql .快速删除 yum remove mysql mysql-serv ...
- springboot获取配置文件中的内容
代码: GrilApplication.java @SpringBootApplication public class GrilApplication { public static void ma ...
- Android 数据库 ObjectBox 源码解析
一.ObjectBox 是什么? greenrobot 团队(现有 EventBus.greenDAO 等开源产品)推出的又一数据库开源产品,主打移动设备.支持跨平台,最大的优点是速度快.操作简洁,目 ...