从零开始制作自己的yolov5数据集
前言
图片的重命名,旋转,镜像等
anaconda labelimg
提示:以下是本篇文章正文内容,下面案例可供参考
一、图片(扩充数据集以及文件重命名)
首先你要拍一些图片
// A code block
数据集的扩充
// An highlighted block
# # -*- coding: UTF-8 -*-
# import glob
# from PIL import Image
# from PIL import ImageFile
# ImageFile.LOAD_TRUNCATED_IMAGES = True
#
# #对图像读取,进行旋转和翻转处理
# imageNum = glob.glob(r"D:/数据集/数据集一/*.jpg")
# print(len(imageNum))
#
# for i in range(len(imageNum)):
# im = Image.open(''+imageNum[i])
# im_rotate_90 = im.rotate(90)
# im_rotate_180 = im.rotate(180)
# im_rotate_transpose_LEFT_RIGHT= im.transpose(Image.FLIP_LEFT_RIGHT)
# im_rotate_transpose_TOP_BOTTOM = im.transpose(Image.FLIP_TOP_BOTTOM)
# if i <10:
# #im.save('F:\\数据集\\数据集一编号\\0000' + i.__str__() + '.png')
# im_rotate_90.save('D:\\数据集\\数据集一旋转90度\\0000'+i.__str__()+'.jpg')
# im_rotate_180.save('D:\\数据集\\数据集一旋转180度\\0000' + i.__str__() + '.jpg')
#
# im_rotate_transpose_LEFT_RIGHT.save('D:\\数据集\\数据集一左右翻转\\0000' + i.__str__() + '.jpg')
#
# elif i>=10 and i<100:
# #im.save('F:\\数据集\\数据集一编号\\000' + i.__str__() + '.png')
# im_rotate_90.save('D:\\数据集\\数据集一旋转90度\\000' + i.__str__() + '.jpg')
# im_rotate_180.save('D:\\数据集\\数据集一旋转180度\\000' + i.__str__() + '.jpg')
#
# im_rotate_transpose_LEFT_RIGHT.save('D:\\数据集\\数据集一左右翻转\\000' + i.__str__() + '.jpg')
#
# elif i>=100 and i<1000:
# #im.save('F:\\数据集\\数据集一编号\\00' + i.__str__() + '.png')
# im_rotate_90.save('D:\\数据集\\数据集一旋转90度\\00' + i.__str__() + '.jpg')
# im_rotate_180.save('D:\\数据集\\数据集一旋转180度\\00' + i.__str__() + '.jpg')
#
# im_rotate_transpose_LEFT_RIGHT.save('D:\\数据集\\数据集一左右翻转\\00' + i.__str__() + '.jpg')
#
#
#
#
#
# #对图像进行读取重新编号,放在同一个文件夹中
# imageNum1 = glob.glob(r"D:/数据集/数据集一/*.jpg")
# imageNum3 = glob.glob(r"D:/数据集/数据集一旋转90度/*.jpg")
# imageNum4 = glob.glob(r"D:/数据集/数据集一旋转180度/*.jpg")
# imageNum5 = glob.glob(r"D:/数据集/数据集一左右翻转/*.jpg")
#
# print(len(imageNum1))
#
# imageNumALL = imageNum1+imageNum3+imageNum4+imageNum5
# print(len(imageNumALL))
# print(imageNumALL)
# for i in range(len(imageNumALL)):
# im = Image.open(''+imageNumALL[i])
# if i < 10:
# im.save('D:\\数据集\\数据集一集合\\0000' + i.__str__() + '.jpg')
# elif i>=10 and i<100:
# im.save('D:\\数据集\\数据集一集合\\000' + i.__str__() + '.jpg')
# elif i>=100 and i<1000:
# im.save('D:\\数据集\\数据集一集合\\00' + i.__str__() + '.jpg')
# elif i >= 1000 and i < 10000:
# im.save('D:\\数据集\\数据集一集合\\0' + i.__str__() + '.jpg')
下面展示一些 内联代码片
。
// A code block
图片的重命名
// An highlighted block
#图片批量重命名
import glob
from PIL import Image
import os
path = "D:\\desk\\数据集\\*.*"
savepath = "D:\\desk\\数据集1\\"
i = 1
for jpgfile in glob.glob(path):
img = Image.open(jpgfile)
img_name = os.path.basename(jpgfile)
img_id = img_name.split('.')[0]
img.save(savepath+ format(str(i), '0>5s') +".jpg")
i += 1
二、环境配置
1.安装anaconda
下面两个都比较好
链接: https://blog.csdn.net/qiqi1220/article/details/121647936.
2.配置labelimg
2.1下载labelimg
链接: https://gitcode.net/mirrors/tzutalin/labelimg?utm_source=csdn_github_accelerator.
下载之后解压缩,放在电脑任意盘(解压缩之后的文件夹不能删)
2.2 安装相关依赖
使用cd 命令进入刚刚下载解压的labelimg文件夹
进入这个目录后按照顺序输入这几个命令
conda install pyqt=5
pyrcc5 -o resources.py resources.qrc
python labelImg.py
之后就打开了
三.标注
全部流程
一个一个全部标注
标完之后保存
保存为.xml文件
注意标的时候要尽量把物体全部框住
总结
学习总结,仅供参考。
如有错误,多多指教