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 #249

Closed
wants to merge 0 commits into from

Conversation

soumik12345
Copy link

Modifications added in this PR:

  • Download the model automatically from the release based on an alias
  • Download the image for inference from a URL
  • tools/infer.py and yolov6/core/inferer.py formatted using black
  • Added support for logging inference results to Weights & Biases. This enables us to:
    • easily log our results to Weights & Biases
    • explore and analyze the bounding box predictions using a simple and intuitive user interface.

Link to Weights & Biases Dashboard: https://wandb.ai/geekyrakshit/yolov6-integration-test

wandb-dashboard

@Chilicyy
Copy link
Collaborator

Thanks for your PR. There are some conflicts to be resolved. Could you git pull our latest code and pull requests again?

@soumik12345
Copy link
Author

Sure, I will resolve the conflicts.

@soumik12345
Copy link
Author

Hi @Chilicyy I raised a new PR (#543) for this.

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