生成lmdb文件和均值文件

最新推荐文章于 2024-07-17 15:12:53 发布

DANGEROUSW

最新推荐文章于 2024-07-17 15:12:53 发布

阅读量726

点赞数 1

分类专栏： caffe 文章标签： caffe 生成lmdb文件生成均值文件

本文链接：https://blog.csdn.net/qq_37254336/article/details/80830132

版权

本文详细介绍了如何准备训练数据和测试数据，生成标签文件，然后利用Caffe创建lmdb文件，并计算均值文件，用于深度学习模型的训练。主要涉及的工具有convert_imageset和compute_image_mean。

摘要由CSDN通过智能技术生成

一、首先准备好训练数据和测试数据，每一类图片放到一个文件夹中

二、生成标签文件：这个代码对应的是七分类

"<span style=""font-family:Arial;font-size:18px;"">"
"<span style=""font-size:18px;"">"
"<span style=""font-size:18px;"">"
import os

def IsSubString(SubStrList,Str):
flag=True
for substr in SubStrList:
if not(substr in Str):
flag=False

return flag
def GetFileList(FindPath,FlagStr=[]):
FileList=[]
FileNames=os.listdir(FindPath)
if len(FileNames)>0:
for fn in FileNames:
if len(FlagStr)>0:
if IsSubString(FlagStr,fn):
fullfilename=os.path.join(FindPath,fn)
FileList.append(fullfilename)
else:
fullfilename=os.path.join(FindPath,fn)
FileList.append(fullfilename)

if len(FileList)>0:
FileList.sort()

return FileList

train_txt=open('train.txt','w')

imgfile=GetFileList('/home/ths/6_operation_32_data/database/train/train_bilinear/')
for