voc数据集处理

python 对txt中每行内容进行批量替换

python 从txt文件查找含某字段的行,并将这些行名称的图像提取出来另存

Python 统计文件行数

将voc数据集中bottle_train.txt文件中的正图片提取出来另存到文件夹中

#-*- coding:UTF-8 -*-
import re
from PIL import Image

f1 = open('/home/wei/MyPaperCode/VOC2012/ImageSets/Main/bottle_train.txt')
f2 = open('bottle_train.txt', 'w+')
for line in f1.readlines():
    if re.findall(' 1', line):#查找“空格1”的行
        f2.write(line)#将查找到的行写入f2
f1.close()
f2.close()
f3 = open('bottle_train.txt', 'r')
for line2 in f3.readlines():
    line3 = line2[:-4]#读取每行去掉后四位的数
    im = Image.open('JPEGImages/{}.jpg'.format(line3))#打开该路径下line3记录的文件名
    im.save('myDataSet/bottle/{}.jpg'.format(line3))#把文件夹中指定的文件名的图片另存到该路径下
f3.close()

将voc数据集中bottle_train.txt文件中的正图片的标注提取出来另存到文件夹中,此方法也可用于上面的图片另存

#-*-coding:utf-8 -*-
import os
import shutil

f1 = open('bottle_train.txt', 'r')
for line in f1.readlines():
    line1 = line[:-4]
    f2 = ('/home/wei/MyPaperCode/VOC2012/Annotations/{}.xml'.format(line1))
    shutil.copy(f2, 'myDataSet/Annotations/')

Pascal VOC 数据集下载与转换

YOLOv2训练自己的数据集(VOC格式)

将数据集做成VOC2007格式用于Faster-RCNN训练

YOLOv3 实练(以VOC2007、2012数据集为例)

猜你喜欢

转载自blog.csdn.net/infinita_LV/article/details/82821497