我是python的新手,对python不是很熟悉。 我想修改的在线代码,以转换我的图像数据到lmdb形式。你知道吗
我给出了src和dst的根,如下所示: 路径\u src='F:\caffe windows\caffe windows\data\sift flow\test' path_dst='F:\caffe windows\caffe windows\data\sift flow\testlmdb'
但是在我运行代码之后,我得到了一个错误。看来我的路错了?有人能帮忙吗?我也附上代码。你知道吗
import os
import numpy as np
from scipy import io
import lmdb
import caffe
from PIL import Image
NUM_IDX_DIGITS = 10
IDX_FMT = '{:0>%d' % NUM_IDX_DIGITS + 'd}'
print '1111'
paths_src = 'F:\\caffe-windows\\caffe-windows\\data\\sift-flow\\test\\'
path_dst = 'F:\\caffe-windows\\caffe-windows\\data\\sift-flow\\testlmdb'
print '2222'
def img_to_lmdb(paths_src,path_dst):
in_db = lmdb.open(path_dst, map_size=int(1e9))
with in_db.begin(write=True) as in_txn:
for in_idx, in_ in enumerate(paths_src):
print 'img:::'+str(in_)
# load image:
# - as np.uint8 {0, ..., 255}
# - in BGR (switch from RGB)
# - in Channel x Height x Width order (switch from H x W x C)
im = np.array(Image.open(in_)) # or load whatever ndarray you need
im = im[:,:,::-1]
im = im.transpose((2,0,1))
im_dat = caffe.io.array_to_datum(im)
in_txn.put('{:0>10d}'.format(in_idx), im_dat.SerializeToString())
in_db.close()
img_to_lmdb(paths_src,path_dst)
print '3333'
我不熟悉此库,但您的问题始于此行-
for in_idx, in_ in enumerate(paths_src):
因为paths\u src是一个字符串,是iterable,所以内置函数enumerate只需遍历该字符串中的每个字符,因此第一项将是字母“F”,而不是整个路径。 您应该做的是将路径定义为一个列表。所以不是:
你应该做:
希望我能帮上忙。你知道吗
相关问题 更多 >
编程相关推荐