Reconhecimento de sinais de trânsito em ambiente rodoviário real - detecção de
ponto-alvo com rede neural autotreinada
Tutorial em vídeo 42 - treinamento de detecção de alvo por rede neural, sinais de trânsito em ambiente rodoviário real: https://singtown.com/learn/50918/
O código, o modelo e o conjunto de dados estão todos em: https://github.com/SingTown/Traffic-Sign-FOMO
Passos em chinês: https://github.com/SingTown/Traffic-Sign-FOMO/blob/main/README-CN.md
Foram treinados dois modelos ao todo
O primeiro modelo: a resolução durante o treinamento foi de 128*128, usando transfer learning com MobilNetV2 0.35, S1 Score de 92%, taxa de quadros de execução de cerca de 9 quadros, e só pode ser executado no OpenMV4 Plus. (O OpenMV4 não tem memória suficiente)
O segundo modelo: resolução de 96*96 durante o treinamento, usando transfer learning com MobilNetV2 0.35, S1 Score de 91%, taxa de quadros de execução de cerca de 12 quadros, uso de pico de RAM de 244KB, pode ser executado no OpenMV4 e no OpenMV4 Plus.
Site de treinamento online do parceiro OpenMV, EdgeImpulse: https://www.edgeimpulse.com/
Observação:
- Há dois modelos ao todo. Se for o OpenMV4, você só pode usar o modelo pequeno treinado com resolução de 96*96. O Plus pode usar os dois modelos.
- O conjunto de dados contém cinco pastas, incluindo Proibido Estacionar x, Proibido Buzinar n, Limite de Velocidade 80, Proibido Entrar- e Imagens de Teste. No modelo e no tutorial reais, usamos apenas Proibido Estacionar x, Proibido Buzinar n, Limite de Velocidade 80 e Imagens de Teste. Não há conjunto de dados rotulado e treinado para Proibido Entrar-. Você pode escolher se quer rotular e treinar o Proibido Entrar-.
- Eu só rotulei os três alvos Proibido Estacionar x, Proibido Buzinar n e Limite de Velocidade 80. Não rotulei nem treinei outros sinais no conjunto de dados (como Limite de Velocidade 60, Limite de Velocidade 30, Proibido Entrar, etc.). Sinais muito parecidos, como Limite de Velocidade 80 e Limite de Velocidade 60, podem ser identificados incorretamente. Você também pode adicionar anotações e treinar novos alvos, como Limite de Velocidade 60, etc.
- Agradecimentos - Fonte do conjunto de dados de sinais de trânsito em ambiente rodoviário real da China (Instituto de Automação, Academia Chinesa de Ciências): http://www.nlpr.ia.ac.cn/pal/trafficdata/detection.html