在Python中指定文件路径时出错

2024-05-17 04:04:05 发布

您现在位置:Python中文网/ 问答频道 /正文

我是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'

enter image description here


Tags: pathinfromimportsrclmdbdatawindows
1条回答
网友
1楼 · 发布于 2024-05-17 04:04:05

我不熟悉此库,但您的问题始于此行-
for in_idx, in_ in enumerate(paths_src):

因为paths\u src是一个字符串,是iterable,所以内置函数enumerate只需遍历该字符串中的每个字符,因此第一项将是字母“F”,而不是整个路径。 您应该做的是将路径定义为一个列表。所以不是:

paths_src = 'F:\\caffe-windows\\caffe-windows\\data\\sift-flow\\test\\'

你应该做:

paths_src = ['F:\\caffe-windows\\caffe-windows\\data\\sift-flow\\test\\']

希望我能帮上忙。你知道吗

相关问题 更多 >