Home › Fóruns › Fórum Detecção de Objetos com YOLO, Darknet, OpenCV e Python › Salvando durante o treinamento
Marcado: (YOLOv8)
- Este tópico contém 1 resposta, 2 vozes e foi atualizado pela última vez 10 meses atrás por
Gabriel Alves.
Visualizando 2 posts - 1 até 2 (de 2 do total)
- AutorPosts
- 15 de abril de 2024 às 22:00 #44475
- Quantas épocas são recomendas para treinar usando yolo v8?
- Há alguma forma de salvar os melhores pesos no google drive enquanto o sistema está realizando treinamento?
17 de abril de 2024 às 19:21 #44488Olá Rafael!
- Para o YOLOv8 não há uma recomendação pois depende muito do dataset e de seu tamanho ou suas características, mas uma dica geral é começar com pelo menos uns 300 epochs. Em seguida, verifique se não ocorre o overfitting, caso não ocorra continue treinando por mais tempo. Aumente para 500, 700, 1000, e assim sucessivamente. Ou seja, enquanto não ocorrer o overfitting você pode continuar sempre treinando mais. E observe a métrica mAP durante o treinamento, quando notar que ela estabilizar e não aumentar mais após algumas épocas então treinou o suficiente (porém, oscilar um pouco entre as epochs é normal).
- Para salvar direto no Drive, você pode usar o parâmetro “project” da função train().
Com ele, você pode mudar o local onde deseja que os pesos treinados sejam salvos. Se você passar um caminho que seja dentro de /content/gdrive/MyDrive/ (igual como foi demonstrado nas aulas de treinamento do YOLOv4) então os pesos atualizados serão salvos dentro do Drive. E os melhores pesos serão aqueles com “-best” no nome.
Exemplo de código:
project = ‘/content/gdrive/MyDrive/treinamentoYOLO’ # diretorio drive
resultados = model.train(data=arquivo_config, epochs=10, imgsz=640, name=’yolov8s_modelo’, project=project) - AutorPosts
Visualizando 2 posts - 1 até 2 (de 2 do total)
- Você deve fazer login para responder a este tópico.