-
Notifications
You must be signed in to change notification settings - Fork 1
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
Add YOLOv4 to studio #120
Add YOLOv4 to studio #120
Conversation
…ys show icon to upload
…nto object-detection
…um to support WH_ARCH_YAML setting
…tion & use modelID in endpoint invocation
- adds a component to open live webcam - hit endpoint - draws bounding boxes WIP: has errors
…up API call and detection handling
@bgoelTT have any screenshots or a walkthrough of the frontend for the object detection demo? |
Yes, please see this video https://drive.google.com/file/d/1Jxvbvl79YtoktRYD3GLSyRQC-RiLXpDv/view?usp=share_link It demonstrates the two modes for the new |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Post testing feedback
- Dropdown Menu:
- Update the
yolov4
option in the dropdown to use capital letters (YOLOV4
) to maintain consistency with naming conventions.
- Update the
- Tooltip on Models Deployed Page:
- Modify the tooltip for the "Object Detection" button.
- Webcam Cleanup:
- We would need to implement a mechanism to stop or clean up the webcam when:
- The user navigates away from the "Start Webcam" page.
- The user clicks the "Stop" button.
- Currently, the green light on my MacBooks remains on, indicating the webcam is still active even after navigation. This would need to be addressed for better resource management and user privacy.
- We would need to implement a mechanism to stop or clean up the webcam when:
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pending @anirudTT changes, looks good.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
* Initial commit - add object detection route * Add package-lock.json * Add two-column object detection component * Add new layout and component structure * Use Aceternity UI file picker * adds tabs to control menu * modifies to move webcam to main component * adds webcam component * add the react package for webcam util * add shadcnn tabs ui component * modifies file upload to show last uploaded file + color change + always show icon to upload * Fix containing element scroll and z-stack * Add overflow scroll to main component * Allow images to assume full width of ObjectDetectionComponent * Add YoloV4 model config to backend API * Create new object-detection endpoint & expand DeviceConfigurations enum to support WH_ARCH_YAML setting * Add ModelType enumeration in frontend to faciliate conditional navigation & use modelID in endpoint invocation * WIP add components to support: - adds a component to open live webcam - hit endpoint - draws bounding boxes WIP: has errors * draw box on image * remove * Optimize real-time object detection to prevent frame backlog * Ensure webcam stops completely when stop button is clicked + layout changes * ts fixes * Fix aspect ratio of video container to 4:3 * Fix navigation and add <img> to SourcePicker component - TODO - wire up API call and detection handling * Refactor inference API call and UI * Fix UI bugs * Add API authentication to YOLOv4 backend * Address PR comments --------- Co-authored-by: Anirudh Ramchandran <[email protected]>
This PR adds frontend and backend support to TT-Studio. It fully implements:
Future improvements