You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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
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.
The text was updated successfully, but these errors were encountered: