Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Convert weights per model into bin files #133

Open
1 of 5 tasks
epierotti3 opened this issue Apr 1, 2023 · 0 comments
Open
1 of 5 tasks

Convert weights per model into bin files #133

epierotti3 opened this issue Apr 1, 2023 · 0 comments

Comments

@epierotti3
Copy link
Contributor

epierotti3 commented Apr 1, 2023

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.

@epierotti3 epierotti3 added the enhancement New feature or request label Apr 1, 2023
@epierotti3 epierotti3 added size: 2 pt 1-3 hours of work role: data science / ML engineers and removed enhancement New feature or request labels Apr 5, 2023
@epierotti3 epierotti3 moved this to To-Do in TrashAI Apr 5, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
Status: To-Do
Development

No branches or pull requests

1 participant