Граф коммитов

5 Коммитов

Автор SHA1 Сообщение Дата
Nat Kershaw (MSFT) 5c53aaad62
Update Yolov8 tutorial with arguments (#658) 2024-02-21 20:11:45 -08:00
Scott McKay da9c01d3e1
Add support for YOLO v8 Pose post-processing. (#605)
* Add support for YOLO v8 Pose post-processing.

The output has additional values in the 'mask' data for the keypoints.

- Update the post processing steps to support extracting and scaling the keypoints.
- Simplify the existing step to split out the boxes and scores by using a basic Split operator if there is no confidence score for a bounding box to apply to the class scores.
  - Confidence score for a bounding box is YOLO versions prior to 8.
- Update existing tests

TODO: Add unit tests for new Steps. They have been manually validated with the real model for now.

* Changes to support pre-decoded input.

Needs cleanup.

* Support an overall max number of detections as well as per-class detections.

* Expand Identity to support multiple inputs
Fix issue with incorrect score being selected by NMS (was max and not score for selected clas)
Fix TopK usage so result ordering is consistent when it is not used
Add unit tests.

* Update docs and some cleanups

* Use Union
2023-12-20 08:51:29 +10:00
Scott McKay 5d53f91f11
Minor opencv tweaks (#316)
* Separate ops in operators/cv2 that do and do not require codecs so they're easier to include/exclude from a build.
Remove jpeg2000 from opencv file formats. It costs 1MB and is (afaict) not a common format.
Add ability to enable/disable cv2 ops to gen_selectedops.py.

* Remove super resolution pre/post process ops that are no longer needed.

* Replace super resolution e2e tutorial

Co-authored-by: Wenbing Li <10278425+wenbingl@users.noreply.github.com>
2022-12-20 13:47:17 +10:00
shaahji 78d8dd5705 OpenCV Image Decoder & SuperResolution CustomOps 2022-09-30 12:08:38 -07:00
Wenbing Li a563194d41
The README.md doc update for the release (#87)
* Update the README.md for the release.

* polish the code

* add the mobile section

* update the example code in the README.md

* correction on pytorch tutorial
2021-05-07 14:12:25 -07:00