【从零开始制作自己的yolov5数据集】

从零开始制作自己的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.

链接: https://zhuanlan.zhihu.com/p/347990651#:~:text=%E5%AE%89%E8%A3%85Anaconda%201%E3%80%81%E7%9B%B4%E6%8E%A5%E7%82%B9%E5%87%BB%E5%BC%80%E5%A7%8B%E5%AE%89%E8%A3%85%202%E3%80%81%E4%B8%80%E8%88%AC%E6%9D%A5%E8%AF%B4%EF%BC%8C%E5%A4%A7%E5%AE%B6%E7%9A%84%E7%94%B5%E8%84%91%E5%8F%AA%E6%9C%89%E4%B8%80%E4%B8%AA%E8%B4%A6%E6%88%B7%EF%BC%8C%E9%BB%98%E8%AE%A4%E9%80%89%E6%8B%A9Just,me%EF%BC%8C%E5%A6%82%E6%9E%9C%E4%BD%A0%E7%9A%84%E7%94%B5%E8%84%91%E6%9C%89%E5%A4%9A%E4%B8%AA%E8%B4%A6%E6%88%B7%EF%BC%8C%E9%82%A3%E5%B0%B1%E9%80%89%E6%8B%A9All%20Users%203%E3%80%81%E9%80%89%E6%8B%A9%E5%AE%89%E8%A3%85%E5%8C%BA%E5%9F%9F%E6%97%B6%EF%BC%8C%E5%A6%82%E6%9E%9C%E4%BD%A0%E7%9A%84C%E7%9B%98%E8%B6%B3%E5%A4%9F%E5%A4%A7%EF%BC%8C%E5%8F%AF%E4%BB%A5%E7%9B%B4%E6%8E%A5%E9%BB%98%E8%AE%A4%E4%B8%8D%E7%94%A8%E4%BF%AE%E6%94%B9%EF%BC%8C%E9%BB%98%E8%AE%A4%E5%AE%89%E8%A3%85%E5%88%B0C%E7%9B%98%E4%B9%9F%E5%8F%AF%E4%BB%A5%E9%81%BF%E5%85%8D%E5%90%8E%E7%BB%AD%E5%8F%AF%E8%83%BD%E5%87%BA%E7%8E%B0%E7%9A%84%E4%BC%97%E5%A4%9A%E5%B0%8F%E9%97%AE%E9%A2%98%E3%80%82.

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文件

注意标的时候要尽量把物体全部框住

总结

学习总结，仅供参考。
如有错误，多多指教