¿De dónde obtiene los pesos utilizados para transfer learning?
Estoy construyendo un modelo que detecta jabalíes y ciervos para crear un clasificador de animales. (Solo tengo datos de jabalíes y ciervos).
Tengo que usar un modelo en yolov5.
Quiero obtener un buen modelo a través del aprendizaje por transferencia, pero no sé cómo obtener los pesos (pre-entrenados) para clasificar jabalíes y ciervos.
¿Cómo los obtengo?
O simplemente descargo yolov5s.pt
y añado el código –weights yolov5s.pt al entrenar el modelo.
Soy principiante,
si me dices qué estoy haciendo mal. Gracias.
davy.ai
Como principiante, se recomienda utilizar transfer learning con un modelo pre-entrenado para obtener mejores resultados. Para utilizar transfer learning con el modelo YOLOv5, deberás descargar los pesos pre-entrenados.
Los pesos pre-entrenados se pueden encontrar en la página oficial de YOLOv5 en GitHub, en la sección de modelos. Puedes descargar el archivo
.pt
del modelo deseado y utilizarlo como punto de partida para tu propio modelo de aprendizaje profundo.Una vez que hayas descargado los pesos pre-entrenados, puedes utilizar la bandera
--weights
seguida de la ruta al archivo.pt
descargado, al entrenar tu modelo. Por ejemplo, si descargasteyolov5s.pt
, puedes utilizar el comando:python train.py --weights yolov5s
.