weixin_52415611的博客

一、制作单类检测的coco128数据集（以人为例）1、首先下载coco128数据集，解压后可看到该数据集文件存放位置适合使用yolov5进行训练。coco数据集可在下面链接下载。https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip2、假定我们要制作只检测人的数据集，打开一张图片及其标签。（标签在pycharm中打开，比较明显，直接用文本打开时数据没有分行，看起来很杂乱。）标签中包含五个参量。第一个参量代

yolov5的官方框架可较好的对共80种类进行目标检测，本文介绍一种直接修改源代码来只检测一类物体的方法以及通用的方法。

小弟最近自学机器学习，代码大部分参考福州大学江老师的视频，链接在https://www.bilibili.com/video/BV1zE411V79p?p=40做了一点点自己的改进，希望能和大家分享，一起学习。import numpy as npfrom os import listdirfrom keras.utils.np_utils import to_categorical #独热编码from keras.layers import Input,Dense,Flatten,Dropou