[SPIKE] Explore alternative training frameworks
Opened this issue · 0 comments
feydan commented
Currently we use YoloV5.
Can we try training the model on a different framework?
How does it impact accuracy?
Are there other ways to improve accuracy of the model (i.e. more epochs)?
Might it be useful to have a model selector when using TrashAI?
Acceptance Criteria:
- After exploring these models, document the findings in a readme file inside of the repo in the model folder. It would be great to see which models were tested, what the findings were, and ideally a way to reproduce the findings.