---
# System prepended metadata

title: Novo treinamento em YoLov7

---


# Novo treinamento em YoLov7

## Treinamento

Foi utilizado o próprio [guia da Metta](https://gitlab.com/mettainnovations/metta-wiki/-/wikis/YoloV7:-Imagem-Docker-e-prepara%C3%A7%C3%A3o-de-ambiente-de-treinamento)
 para realizar o treinamento.

Porém, seguem os comandos utilizados na Brazil:

Inicializar o container:

<div align="center">
  
[![Project Euler](https://projecteuler.net/profile/arthurvalls.png)](https://projecteuler.net/progress=arthurvalls)
</div>
 
  ```
sudo docker exec -it yolov7-ventura bash
```
Entrar no diretório onde o script para treinamento se encontra:
```
cd ../yolov7/
```

Rodar o script:

```
python train.py \
--epochs 500 \
--workers 4 \
--device 0 \
--batch-size 32 \
--data data/ventura-secundario-tiny.yaml \
--img 640 640 \
--cfg cfg/training/yolov7_ventura-secundario-tiny.yaml \
--weights 'yolov7-tiny.pt' \
--name yolov7_tiny_ventura_secundario \
--hyp data/hyp.scratch.tiny.yaml
```

OBS: Todas as flags acima foram especificadas no antigo relatório para treinamento em yolov7.

## Resultados

Foram treinadas duas redes da tiny da **Ventura (primária e secundária)**. Datasets podem ser encontrados aqui: [Dataset primário](https://92ff9ed447.sharepoint.com/:u:/s/Desenvolvimento/ERs-wU2WhFNPkMyDlRDjBCoBgmr9LP9Ccym7AgpVTO9wHg?e=gPUtPa) e [Dataset secundário](https://92ff9ed447.sharepoint.com/:u:/s/Desenvolvimento/EVegvYSN4rhMkH-qbwEtu5IB4-kBa7N12umSEi5EoiwJvw?e=9eRoMt). Foi utilizado um script para realizar a separação do dataset em **train** (70%), **val** (15%) e **test** (15%). O script **divide_dataset.py** pode ser encontrado [aqui]().


| Rede |  Épocas   | Passos por Época | Tempo de Treinamento | mAP |
| :---:   | :---: | :---: | :---:  | :---:  |
| Primária | 500   | 74   | 19h | 99.3%  |
| Secundária | 500 | 237 | 14.7h | 94.2% |

### Resultados da rede primária:

![](https://i.imgur.com/7Y3rBFX.jpg)

![](https://i.imgur.com/xvvO3oT.jpg)

### Resultados da rede secundária:

![](https://i.imgur.com/kdV3d4q.jpg)

![](https://i.imgur.com/ps8bT2y.jpg)




## Comparações

### Inferência em Python yolov4

![](https://media.giphy.com/media/v1.Y2lkPTc5MGI3NjExNDFjODhiMzMwMWY4OTdjZTAxZDZiYjVmYjMxYTg0MzY0NTZmYzc4MCZjdD1n/sDH62aIhq1FP4nAi0z/giphy-downsized-large.gif)


![](gif_ventura2.mp4)
