Implementando um detector de objetos (Single-Shot Detection) do zero mesmo.

Informações importantes

Um pouco de teoria

Modelo

AdrianoNet: nossa arquitetura de “brinquedo”.

Dataset

Funcoes auxiliares para pré-processamento dos dados.
Classe dataset.

Trabalhando com os dados

Funcão letterbox
Funcão de transformacão de imagens.

Treinamento

Código completo: https://github.com/adrianosantospb/detector_classificaodor_do_zero/blob/main/train.py

Testando o modelo treinado

Considerações finais

--

--

--

Senior Computer Vision Data Scientist at Conception Ro-Main (Quebec — CA). DSc in Computer Science. MTAC Brazil. https://github.com/adrianosantospb

Love podcasts or audiobooks? Learn on the go with our new app.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Adriano A. Santos

Adriano A. Santos

Senior Computer Vision Data Scientist at Conception Ro-Main (Quebec — CA). DSc in Computer Science. MTAC Brazil. https://github.com/adrianosantospb

More from Medium

Neural Style Transfer

Paper Review: “YOLOX: Exceeding YOLO Series in 2021”

Custom Dataset with Dataloader in Pytorch

Semi-Supervised Learning | Pseudo Labeling Custom Dataset with YOLOv4