关于自己训练tank

最新推荐文章于 2024-11-29 16:57:46 发布

原创最新推荐文章于 2024-11-29 16:57:46 发布 · 176 阅读

0 ·

CC 4.0 BY-SA版权

本文探讨了使用ImageNet数据集时图片格式的灵活性，详细介绍了如何利用os.walk与os.listdir遍历文件夹并批量修改图片后缀名，同时提供了将图片信息写入txt文件的实用方法。

1，如果使用imagenet是不是一定要用“.jpeg”结尾的图片呢？不一定

参考https://blog.youkuaiyun.com/legend_hua/article/details/80046578

2、关于遍历问题：（还差一个rename的问题）

1，关于os.walk与os.listdir的区别。os.listdir只检测当前路径下面的文件，不会进一步进入子文件。但是os.walk是会把该文件夹下面的所有文件都给搜索出来，包括子文件夹中的文件。所以，修改tank的后缀名可以如下：

参考：https://www.jb51.net/article/46260.htm

os.path.splitext('path'),

os.rename()的注意要点：https://zhidao.baidu.com/question/1576885978232165980.html

实现一，将220个文夹夹下面的图片的名字的后缀由.jpg改为jpeg

解决方案：

# -*- coding: utf-8 -*-
import os,sys
imgpath='/home/alice/caffe/data/tank/m_tank_val'
filelist=[]
for imgname in os.listdir(imgpath):
    #print(imgname)

    #https://blog.youkuaiyun.com/u012193416/article/details/77247432
    fullfilename=os.path.join(imgpath,imgname)#连接目录与文件名或目录,
    filelist.append(fullfilename)
    l=len(filelist)
#print filelist
#print l

for m in range(0,l):
    for tupian in os.listdir(filelist[m]):
        portion = os.path.splitext(tupian)
        # print portion
        if portion[1] == ".jpg":
            # 重新组合文件名和后缀名
            newname = portion[0] + ".jpeg"
            #print newname
            
            #os.renames(tupian,newname)
            os.rename(os.path.join(filelist[m], tupian), os.path.join(filelist[m], newname))

经证明以上程序运行，可以实现一个文件夹下面多个子文件夹中图片的后缀名的修改

实现二：考虑将220个文件夹下面的图片的名字写入到txt里面，并且将相应分类写在名字的后面，中间加上一个空格

# -*- coding: utf-8 -*-
import os,sys
doc=open('/home/b101/caffe/data/tank/label.txt','w')
imgpath='/home/b101/caffe/data/tank/m_tank_val'
filelist=[]
for imgname in os.listdir(imgpath):
    fullfilename = os.path.join(imgpath, imgname)  # 0-220
    filelist.append(fullfilename)
    l=len(filelist)#多少个分类
    for m in range(0, l):#对每一分类进行操作
        for tupian in os.listdir(filelist[m]):#获取每一个分类里面的图片的名字
            print(tupian)
            print(tupian," ",imgname,file = doc)
doc.close()

但是，denny其实给出了一个更好的方案：

参考连接：https://www.cnblogs.com/denny402/p/5083300.html

#!/usr/bin/env sh
DATA=data/re/
MY=examples/myfile

echo "Create train.txt..."
rm -rf $MY/train.txt
for i in 3 4 5 6 7
do
find $DATA/train -name $i*.jpg | cut -d '/' -f4-5 | sed "s/$/ $i/">>$MY/train.txt
done
echo "Create test.txt..."
rm -rf $MY/test.txt
for i in 3 4 5 6 7
do
find $DATA/test -name $i*.jpg | cut -d '/' -f4-5 | sed "s/$/ $i/">>$MY/test.txt
done
echo "All done"

以下是尝试过的方案

（1）方案一：问题在于os.walk（）貌似与os.path.splitext（）不兼容

import os
files = os.walk(".jpg")

for filename in files:
    portion = os.path.splitext(filename)
    if portion[1] == ".jpg":
        # 重新组合文件名和后缀名
newname = portion[0] + ".jpeg"
os.rename(filename,newname)

（2）方案二：问题在于os.walk（）貌似与os.path.splitext（）不兼容

参考：https://www.cnblogs.com/xingh/p/6937353.html

import os

files = os.walk("D:\QTDownloadRadio\QTDownloadRadio - 副本")#列出当前目录下所有的文件

#这里使用os.walk.后面的os.path.splitext()就出现问题了

for filename in files:
portion = os.path.splitext(filename)#分离文件名字和后缀
print(portion)

if portion[1] ==".jpg":#根据后缀来修改,如无后缀则空
newname = portion[0]+".jpeg"#要改的新后缀
os.rename(filename,newname)

曾经的尝试：

2，os.listdir(path).这里的path一定要是“具体的路径吗” path1=path+“/”+file这种合成的不可以吗？

os.path.join(path,each_file)

3，for file in os.listdir(data_base_dir): # file为current_dir当前目录下图片名 write_name = file # 图片路径 + 图片名 + 标签 file_list1.append(write_name) # 将write_name添加到file_list列表最后 number_of_lines = len(file_list1) # 列表中元素个数

def imgpath1(imgpath): filelist=[] for imgname in os.listdir(imgpath): #print(imgname)

#https://blog.youkuaiyun.com/u012193416/article/details/77247432
fullfilename=os.path.join(imgpath,imgname)#在上一级的路径下加上文件的名字

filelist.append(fullfilename)

l=len(filelist)

for tupian in os.listdir(filelist[l-1]):

#<后缀重命名的程序>

  portion = os.path.splitext(tupian)
if portion[1] == ".jpg":
        # 重新组合文件名和后缀名
        newname = portion[0] + ".jpeg"
        os.rename(filename,newname)
3、如何查看一个网络对训练的图片的要求，比如

在matlab中，构建的网络中有：imageInputLaper([28 28 1])

#说明图片的尺寸是28 * 28 *1（灰度图像，仅一个通道）