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

Weights & Biases integration for inference #543

Open
wants to merge 5 commits into
base: main
Choose a base branch
from

Conversation

soumik12345
Copy link

@soumik12345 soumik12345 commented Oct 11, 2022

Modifications added in this PR:

  • Download the model automatically from the release based on the model file
  • Download the image for inference from a URL
  • Added support for logging inference results to Weights & Biases.

Sample Workspace: https://wandb.ai/geekyrakshit/yolov6-integration-test

Sample Usage:

python tools/infer.py --weights yolov6s.pt --source data/images/ --wandb_project <wandb-project-name> --wandb_entity <wandb-entity-name>

…nload model for inference + option to infer on an image from a url
@soumik12345
Copy link
Author

cc: @Chilicyy

@Chilicyy
Copy link
Collaborator

Thanks for your contribution! We'll check and verify the code asap.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants