code4sac/trash-ai

Convert weights per model into bin files

epierotti3 opened this issue · 0 comments

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.