Oii, Lucas, tudo bem contigo?
Você pode usar a PlayTorch. Inclusive, um tutorial oficial da PlayTorch para classificação de imagens está disponível aqui.
Se outra dúvida surgir, estamos disponíveis. :)
Abraços!
Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓.
