code4sac/trash-ai

[SPIKE] Explore alternative training frameworks

Opened this issue · 0 comments

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.