深度学习(七十)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 ...
随机推荐
- 20145321 《Java程序设计》课程总结
20145321 <Java程序设计>课程总结 读书笔记链接汇总 第一周读书笔记 第二周读书笔记 第三周读书笔记 第四周读书笔记 第五周读书笔记 第六周读书笔记 第七周读书笔记 第八周读书 ...
- Mysql 默认编码问题
新安装的数据库默认编码是 latin1 +--------------------------+----------------------------+ | Variable_name | Valu ...
- python 使用getopt 获取配置参数
在工程中特别是稍微大一点的项目基本上都会用到配置,就会涉及到配置文件的读取,配置参数的读取. 常用的解析配置文件的是configParser,解析命令行参数的则为getopt. getopt的参数可以 ...
- UVa 11324 最大团(强连通分量缩点)
https://vjudge.net/problem/UVA-11324 题意:给一张有向图G,求一个结点数最大的结点集,使得该结点集中任意两个结点u和v满足,要么u可以到达v,要么v可以达到u. 思 ...
- 淘海外分发Job 多线程demo
using System;using System.Collections.Generic;using System.Configuration;using System.Diagnostics;us ...
- Android开发-网络通信1
使用 org.apache.http.client.HttpClient; 一开始从官网下载HttpClient 4.5:http://hc.apache.org/downloads.cgi ,解压之 ...
- nodejs中mysql断线重连
之前写了个小程序Node News,用到了MySQL数据库,在本地测试均没神马问题.放上服务器运行一段时间后,偶然发现打开页面的时候页面一直处于等待状态,直到Nginx返回超时错误.于是上服务器检查了 ...
- 用if写一个备份mysql的脚本
#!/bin/bash # 备份数据库 BAK_DIR=/data/backup/`date +%Y%m%d` MYSQLDB=dexin MYSQLUSER=root MYSQLPW=123456 ...
- MooseFS技术详解
原文 http://www.tuicool.com/articles/vQvEZ3y MooseFS是一款具有冗余容错功能的分布式文件系统.它把数据分散在多台服务器上,确保一份数据多个备份副本,对外提 ...
- python学习笔记(字典乱码)
博主总结下 python中字典中包含中文时,使用过程中出现乱码 json.dumps(params, encoding="UTF-8", ensure_ascii=False) p ...