caffe python读取lmdb文件

原创 2018年04月17日 11:39:02
一般用caffe训练模型的时候,都会首先把数据写入数据库,lmdb/leveldb, 但如果写入之后想check一下是否写对了,怎么去读lmdb文件呢?
这里用python代码读取lmdb文件,查看其中的内容,看是否符合预期。


代码如下:


-Python 代码
01
import caffe
02
import lmdb
03
 
04
lmdb_env = lmdb.open('train_lmdb') #caffe保存lmdb文件的目录
05
lmdb_txn = lmdb_env.begin()
06
lmdb_cursor = lmdb_txn.cursor()
07
datum = caffe.proto.caffe_pb2.Datum()
08
 
09
for key, value in lmdb_cursor:
10
    datum.ParseFromString(value)
11
    label = datum.label
12
    data = caffe.io.datum_to_array(datum)
13
    print label, data


这里导入了lmdb和caffe两个库,解释一下如何导入


1、lmdb,需要用pip安装lmdb,进入root账户,输入 pip install lmdb即可(在其它用户下,输入sudo命令无法安装成功)


2、import caffe, 安装好caffe后,在bashrc中,添加环境变量, export PYTHONPYTH="xxx/caffe-master/python", 再source ~/.bashrc


import caffe的时候,会报一个错,是因为没有编译pycaffe,在caffe-master下,make pycaffe即可。
版权声明: https://blog.csdn.net/b876144622/article/details/79972560

【Python】第二周Python基础(下)

本周课程主要讲解Python的面向对象编程, 异常处理、模块、MySQL数据库、Python的数据库支持以及图形用户界面等技术,最后我们使用阶段案例开发了一个《飞机大战》的游戏。
  • 2018年03月16日 11:34

caffe 数据库LMDB的读写

读写的图片都是灰度图,rgb图类似 一、读数据(图片的channel是2,其实是两张图片): import sys sys.path.insert(0,"../../python") import ...
  • u010668907
  • u010668907
  • 2016-07-05 20:50:56
  • 8829

Caffe中图像写入LMDB(以及多标签)和读取LMDB数据

一个大牛写的教程:《A Practical Introduction to Deep Learning with Caffe and Python》,写的很清晰。 将图像写入LMDB import ...
  • langb2014
  • langb2014
  • 2016-11-01 14:12:54
  • 9496

Caffe学习笔记(二):使用Python生成caffe所需的lmdb文件和txt列表清单文件

转载请注明作者和出处: http://blog.csdn.net/c406495762 Python版本:Python2.7 运行平台:Ubuntu14.04 最后修改时间:2017.4.20...
  • c406495762
  • c406495762
  • 2017-03-29 21:08:31
  • 5494

caffe中用python读写lmdb

环境是ubuntu + python2.7中间加上了gaussian模糊import os import glob import sys import random import math impor...
  • tyu5658
  • tyu5658
  • 2016-08-31 17:37:13
  • 1700

python读取LMDB中的图像

python解析LMDB中的数据。
  • Searching_Bird
  • Searching_Bird
  • 2017-10-17 15:01:10
  • 567

对于caffe当中lmdb文件的制作与读取

part1:制作lmdb文件通过python脚本制作成ldmb文件原料:图片,label,train.txt(图片路径+label)# -*- coding: UTF-8 -*- __author__...
  • mdjxy63
  • mdjxy63
  • 2018-01-02 15:49:01
  • 119

Caffe中LMDB的使用

http://rayz0620.github.io/2015/05/25/lmdb_in_caffe/ 最近做实验,要用Caffe提取CNN特征。官方的extract_feature.bin...
  • haluoluo211
  • haluoluo211
  • 2017-01-14 16:40:42
  • 5104

python lmdb使用

http://www.jianshu.com/p/66496c8726a1 4.1 准备 学习LMDB的时候不禁想到知乎上的提问“有哪些名人长期生活在其他名人的光环下”,说实话感觉查它...
  • haluoluo211
  • haluoluo211
  • 2017-01-14 18:16:04
  • 4332

caffe中lmdb和hdf5读取时的总结

HDF5 写入hdf5  store2hdf5 读取hdf5  H5LT: HDF5 Lite 直接以N*C*H*W写入hdf5 N*C*H*W(H LMD...
  • zykimmy
  • zykimmy
  • 2016-10-27 22:03:07
  • 1777
收藏助手
不良信息举报
您举报文章:caffe python读取lmdb文件
举报原因:
原因补充:

(最多只允许输入30个字)