Inferência local com YOLOv5 em uma webcam
Clone o repositório do Github do Ultralytics YOLOv5 em seu computador e execute detect.py depois de salvar os pesos do modelo de seu treinamento personalizado do modelo YOLOv5 .
- Anote o diretório (pasta) no qual você clona o repositório. Salve o caminho do arquivo, se puder, para facilitar o acesso mais tarde.
- Folha de dicas de comandos do GitHub
Inclua os pesos do modelo na pasta "yolov5" que foi clonada em seu sistema de arquivos (computador) do GitHub.
Executando o modelo de webcam
Depois que os pesos personalizados forem adicionados ao diretório yolov5 em seu computador local, você poderá executar o modelo em sua webcam após mudar para o diretório yolov5 em seu terminal.
Comando básico: execute o YOLOv5 em uma webcam
Como o diretório "yolov5" na imagem de exemplo reside no diretório "mtraor3", e estamos usando os pesos do modelo yolov5s salvos na pasta yolov5 ( /Users/mtraor3/yolov5 ) , o código de exemplo que inserimos em nosso terminal para executar localmente na webcam ficaria assim:
python ./Users/mtraor3/yolov5/detect.py --weights ./Users/mtraor3/yolov5/yolov5s.pt --save_text --save_conf --conf-thres 0.40 --source 0
Salvar previsões e pontuações de confiança em um arquivo .txt com inferência realizada em um limite de confiança de 40%
python path/to/detect.py --weights /path/to/modelweights.pt --save_text --save_conf --conf-thres 0.40 --source 0
Se o arquivo de pesos do modelo foi intitulado "my_custom_weights.pt" então o código se torna:
python ./Users/mtraor3/yolov5/detect.py --weights ./Users/mtraor3/yolov5/my_custom_weights.pt --save_text --save_conf --conf-thres 0.40 --source 0
Argumentos opcionais
Existem mais argumentos opcionais para fazer coisas como alterar o limite de confiança, salvar caixas de previsão cortadas etc.
Mais exemplos:
Salvar as previsões e pontuações de confiança em um arquivo .txt, com inferência em um limite de confiança de 40%.