Directorio de notas
1. Procesamiento de datos
Etiqueta con labelImg
Marque la dirección de descarga del software: 链接：https://pan.baidu.com/s/1xYu2fCGgIp-lVO3fUHpdaQ 提取码：6666
1. Abra el archivo predefined_classes.txt en el directorio de datos y cámbielo a la clase que necesita
2. Abra Dir e importe la carpeta de imágenes que debe marcarse.
3. La carpeta donde se encuentra el archivo generado después de que se complete la marca Cambiar directorio de guardado
4. Cambie el modo al modo yolo como se muestra en la siguiente figura (el octavo icono en el lado izquierdo del modo independiente)
El modo PascalVOC genera archivos .xml y el modo yolo genera archivos .txt
5. Haga clic en la tecla 'w' del teclado para enmarcar la selección
2. Modelo de formación
1. Cree una nueva carpeta al mismo nivel que la carpeta yolov5
2. Cree dos nuevas carpetas
train folder bajo la carpeta yb para almacenar el conjunto de datos de entrenamiento. La
carpeta válida se usa para almacenar el conjunto de datos de prueba.
Cambie el programa en data.yaml
train: ..\yb\train\images
val: ..\yb\valid\images
nc: 2
names: ['fishJ', 'fishL']
train y val son el directorio de almacenamiento de datos
nc: el número de categorías que entrenas
nombres: el nombre de la categoría de entrenamiento
3. Cree dos carpetas nuevas debajo de la carpeta de tren y la carpeta válida. La
carpeta de imágenes se usa para almacenar las imágenes y la
carpeta de etiquetas se usa para almacenar las etiquetas
(labels.cache se genera ejecutando train.py)
4. Cambie el archivo yolov5l.yaml en la carpeta / yolov5 / models (entrené el modelo l)
# parameters
nc: 2 # number of classes
。。。。。。。。。。。。。。。。。（省略一下后面的程序）
Solo cambie la línea nc y cambie 80 al número de categorías que entrena
** 5. ** Ejecute el archivo train.py
python train.py --img 640 --batch 1 --epochs 300 --data ../yb/data.yaml --cfg models/yolov5l.yaml --weights ''
El lote de mi computadora solo se puede ajustar a 1 o se informará un error.
Tres. Errores y soluciones
1. Espacio de memoria insuficiente
OSError: [WinError 1455] 页面文件太小，无法完成操作。 Error loading "E:\anaconda\anaconda\envs\yolov5_GPU\lib\site-packages\torch\lib\caffe2_detectron_ops_gpu.dll" or one of its dependencies.
Solución Cambie el parámetro de lote a 1
y así sucesivamente, si tiene dinero, compre una tarjeta gráfica mejor. . . . .
tiempo de estudio
2021.3.14