Skip to content
This repository has been archived by the owner on Jul 22, 2024. It is now read-only.

Run inference on own picture with externally inputted object labels and bounding boxes #182

Open
aliencaocao opened this issue Jan 17, 2022 · 0 comments

Comments

@aliencaocao
Copy link

I have some images that contain a mixture of seen and unseen object classes. I have my own custom object detection model based on YOLOv5, and it is able to output bounding box and class label and confidence. Is it possible to input these YOLOv5 results into Oscar+ and thus only use the text generation part of Oscar+ to generate a caption of the image? Original image with bounding box can be inputted, but I do not want Oscar+ to do the object detection part as my own model take care of some unseen objects.

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

No branches or pull requests

1 participant