Skip to content

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