技术标签: caffe
前面铺垫了这么多,终于到主题了。
先写一个配置文件conv.protxt
layer {
name: 'MyPythonLayer'
type: 'Python'
top: 'output'
python_param {
module: 'mypythonlayer'
layer: 'MyLayer'
param_str: "{\'data_dir\':\'../../images\',\'num\': 100}"
}
}
然后就是这个python模块怎么写?mypythonlayer.py
#!/usr/bin/env python
#coding=utf-8
import sys
import os
import numpy as np
import os.path as osp
import matplotlib.pyplot as plt
import pylab
from copy import copy
caffe_root = '/home/x/git/caffe/'
sys.path.insert(0, caffe_root + 'python')
import caffe
import numpy as np
import yaml
import numpy as np
from PIL import Image
class MyLayer(caffe.Layer):
def setup(self, bottom, top):
params = eval(self.param_str)
self.data_dir = params['data_dir']
self.num = yaml.load(self.param_str)["num"]
self.cat = '{}/{}.jpg'.format(self.data_dir,'cat')
print "Image:",self.cat
print "Parameter num : ", self.num
def reshape(self, bottom, top):
im = Image.open(self.cat)
in_ = np.array(im, dtype=np.float32)
in_ = in_[:,:,::-1]
self.data = in_.transpose((2,0,1))
top[0].reshape(1, *self.data.shape)
def forward(self, bottom, top):
top[0].data[...] = self.data
print "forward=============================>cat"
print self.data
print type(top[0])
print dir(top[0])
print type(top[0].data)
#print help(top[0].data)
print type(top[0].data[...])
def backward(self, top, propagate_down, bottom):
pass
里面包括几个基本的设置,然后就是参数的读取,我这里添加了两个参数,路径和一个num参数。
基本实现里面有参数读取和图片读取的简单实现。
接下来调用test_python.py
#!/usr/bin/env python
#coding=utf-8
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image
caffe_root = '/home/x/git/caffe/'
import sys
sys.path.insert(0, caffe_root + 'python')
import caffe
import mypythonlayer
sys.path.append("/home/x/git/caffe/examples/test_layer_lb/mypython")
caffe.set_device(0)
caffe.set_mode_gpu()
print "================================="
net = caffe.Net('/home/x/git/caffe/examples/test_layer_lb/mypython/conv.prototxt',caffe.TEST)
#print net.blobs.items()
#print dir(net.blobs)
net.forward()
日志:
WARNING: Logging before InitGoogleLogging() is written to STDERR
I1129 19:11:46.731223 16691 net.cpp:49] Initializing net from parameters:
state {
phase: TEST
}
layer {
name: "MyPythonLayer"
type: "Python"
top: "output"
python_param {
module: "mypythonlayer"
layer: "MyLayer"
param_str: "{\'data_dir\':\'../../images\',\'num\': 100}"
}
}
I1129 19:11:46.731266 16691 layer_factory.hpp:77] Creating layer MyPythonLayer
I1129 19:11:46.731314 16691 net.cpp:91] Creating Layer MyPythonLayer
I1129 19:11:46.731322 16691 net.cpp:399] MyPythonLayer -> output
I1129 19:11:46.744997 16691 net.cpp:141] Setting up MyPythonLayer
I1129 19:11:46.745023 16691 net.cpp:148] Top shape: 1 3 360 480 (518400)
I1129 19:11:46.745028 16691 net.cpp:156] Memory required for data: 2073600
I1129 19:11:46.745035 16691 net.cpp:219] MyPythonLayer does not need backward computation.
I1129 19:11:46.745039 16691 net.cpp:261] This network produces output output
I1129 19:11:46.745045 16691 net.cpp:274] Network initialization done.
=================================
Image: ../../images/cat.jpg
Parameter num : 100
forward=============================>cat
[[[ 49. 50. 47. ..., 30. 36. 45.]
[ 51. 52. 48. ..., 32. 37. 45.]
[ 51. 51. 49. ..., 27. 35. 42.]
...,
[ 15. 13. 13. ..., 156. 162. 164.]
[ 19. 20. 22. ..., 167. 158. 167.]
[ 25. 25. 27. ..., 188. 157. 158.]]
[[ 57. 58. 55. ..., 77. 87. 94.]
[ 57. 58. 56. ..., 82. 89. 96.]
[ 57. 57. 57. ..., 88. 95. 99.]
...,
[ 47. 46. 47. ..., 173. 179. 181.]
[ 53. 56. 58. ..., 184. 175. 184.]
[ 62. 64. 66. ..., 205. 174. 175.]]
[[ 26. 27. 25. ..., 45. 50. 56.]
[ 26. 27. 25. ..., 50. 52. 58.]
[ 26. 26. 26. ..., 52. 55. 60.]
...,
[ 36. 32. 36. ..., 182. 188. 190.]
[ 42. 42. 44. ..., 193. 184. 193.]
[ 46. 49. 51. ..., 214. 183. 184.]]]
<class 'caffe._caffe.Blob'>
['__class__', '__delattr__', '__dict__', '__doc__', '__format__', '__getattribute__', '__hash__', '__init__', '__module__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', '__weakref__', 'channels', 'count', 'data', 'diff', 'height', 'num', 'reshape', 'shape', 'width']
<type 'numpy.ndarray'>
<type 'numpy.ndarray'>
本人水平有点菜,开始一看python的protxt里面的参数有点懵逼,如果看不懂里面的某个东西直接写个文件调试,慢慢就懂了。
a="{\'sbdd_dir\': \'../data/sbdd/dataset\', \'seed\': 1337, \'split\': \'train\', \'mean\': (104.00699, 116.66877, 122.67892)}"
params = eval(a)
print type(())
print params.items()
params_dir=params['sbdd_dir']
print params_dir
文章浏览阅读645次。这个肯定是末尾的IDAT了,因为IDAT必须要满了才会开始一下个IDAT,这个明显就是末尾的IDAT了。,对应下面的create_head()代码。,对应下面的create_tail()代码。不要考虑爆破,我已经试了一下,太多情况了。题目来源:UNCTF。_攻防世界困难模式攻略图文
文章浏览阅读2.9k次,点赞3次,收藏10次。偶尔会用到,记录、分享。1. 数据库导出1.1 切换到dmdba用户su - dmdba1.2 进入达梦数据库安装路径的bin目录,执行导库操作 导出语句:./dexp cwy_init/[email protected]:5236 file=cwy_init.dmp log=cwy_init_exp.log 注释: cwy_init/init_123..._达梦数据库导入导出
文章浏览阅读1.9k次。1. 在官网上下载KindEditor文件,可以删掉不需要要到的jsp,asp,asp.net和php文件夹。接着把文件夹放到项目文件目录下。2. 修改html文件,在页面引入js文件:<script type="text/javascript" src="./kindeditor/kindeditor-all.js"></script><script type="text/javascript" src="./kindeditor/lang/zh-CN.js"_kindeditor.js
文章浏览阅读2.3k次,点赞6次,收藏14次。SPI的详情简介不必赘述。假设我们通过SPI发送0xAA,我们的数据线就会变为10101010,通过修改不同的内容,即可修改SPI中0和1的持续时间。比如0xF0即为前半周期为高电平,后半周期为低电平的状态。在SPI的通信模式中,CPHA配置会影响该实验,下图展示了不同采样位置的SPI时序图[1]。CPOL = 0,CPHA = 1:CLK空闲状态 = 低电平,数据在下降沿采样,并在上升沿移出CPOL = 0,CPHA = 0:CLK空闲状态 = 低电平,数据在上升沿采样,并在下降沿移出。_stm32g431cbu6
文章浏览阅读1.2k次,点赞2次,收藏8次。数据链路层习题自测问题1.数据链路(即逻辑链路)与链路(即物理链路)有何区别?“电路接通了”与”数据链路接通了”的区别何在?2.数据链路层中的链路控制包括哪些功能?试讨论数据链路层做成可靠的链路层有哪些优点和缺点。3.网络适配器的作用是什么?网络适配器工作在哪一层?4.数据链路层的三个基本问题(帧定界、透明传输和差错检测)为什么都必须加以解决?5.如果在数据链路层不进行帧定界,会发生什么问题?6.PPP协议的主要特点是什么?为什么PPP不使用帧的编号?PPP适用于什么情况?为什么PPP协议不_接收方收到链路层数据后,使用crc检验后,余数为0,说明链路层的传输时可靠传输
文章浏览阅读587次。软件测试工程师移民加拿大 无证移民,未受过软件工程师的教育(第1部分) (Undocumented Immigrant With No Education to Software Engineer(Part 1))Before I start, I want you to please bear with me on the way I write, I have very little gen...
文章浏览阅读304次。Thinkpad X250笔记本电脑,装的是FreeBSD,进入BIOS修改虚拟化配置(其后可能是误设置了安全开机),保存退出后系统无法启动,显示:secure boot failed ,把自己惊出一身冷汗,因为这台笔记本刚好还没开始做备份.....根据错误提示,到bios里面去找相关配置,在Security里面找到了Secure Boot选项,发现果然被设置为Enabled,将其修改为Disabled ,再开机,终于正常启动了。_安装完系统提示secureboot failure
文章浏览阅读10w+次,点赞93次,收藏352次。1、用strtok函数进行字符串分割原型: char *strtok(char *str, const char *delim);功能:分解字符串为一组字符串。参数说明:str为要分解的字符串,delim为分隔符字符串。返回值:从str开头开始的一个个被分割的串。当没有被分割的串时则返回NULL。其它:strtok函数线程不安全,可以使用strtok_r替代。示例://借助strtok实现split#include <string.h>#include <stdio.h&_c++ 字符串分割
文章浏览阅读2.3k次。1 .高斯日记 大数学家高斯有个好习惯:无论如何都要记日记。他的日记有个与众不同的地方,他从不注明年月日,而是用一个整数代替,比如:4210后来人们知道,那个整数就是日期,它表示那一天是高斯出生后的第几天。这或许也是个好习惯,它时时刻刻提醒着主人:日子又过去一天,还有多少时光可以用于浪费呢?高斯出生于:1777年4月30日。在高斯发现的一个重要定理的日记_2013年第四届c a组蓝桥杯省赛真题解答
文章浏览阅读851次,点赞17次,收藏22次。摘要:本文利用供需算法对核极限学习机(KELM)进行优化,并用于分类。
文章浏览阅读1.1k次。一、系统弱密码登录1、在kali上执行命令行telnet 192.168.26.1292、Login和password都输入msfadmin3、登录成功,进入系统4、测试如下:二、MySQL弱密码登录:1、在kali上执行mysql –h 192.168.26.129 –u root2、登录成功,进入MySQL系统3、测试效果:三、PostgreSQL弱密码登录1、在Kali上执行psql -h 192.168.26.129 –U post..._metasploitable2怎么进入
文章浏览阅读257次。本文将为初学者提供Python学习的详细指南,从Python的历史、基础语法和数据类型到面向对象编程、模块和库的使用。通过本文,您将能够掌握Python编程的核心概念,为今后的编程学习和实践打下坚实基础。_python人工智能开发从入门到精通pdf