Convert weights per model into bin files
epierotti3 opened this issue · 0 comments
epierotti3 commented
Guidelines
- I agree to follow this project's Contributing Guidelines.
Description
Convert weights per model into bin files
Problem
Most of the current notebooks produce .pt files that contain the trained model weights in them. But these .pt files aren't usable in the application. We need to convert the output of the trained model (the weights) into a usable format (bin files) so that the application can perform identification on uploaded images of trash.
Proposed Solution
- Find the block of code in earlier notebooks that takes in .pt files and converts them .bin files
- Places to look might include Github > Code4Sac/TrashAI/model/build/yolo-5-new ... (last cell of this .ipynb notebook)
- Test out this block of code in the collab notebook to see if the weights were properly generated as .bin files
- If this works, copy this over into the TrashAI repo on Github so it's usable by everyone
Alternatives Considered
If the block of code from the previous notebooks doesn't work, try Github > Code4Sac/TrashAI/notebooks ...one of the notebooks here might have what we need.