HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix RGB to BGR conversion in TFLite example (#18305)

Browse source 
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Mohammed Yasin 2024-12-19 03:00:00 +08:00 • committed by GitHub
parent b55066bd15
commit add15b0ea8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/YOLOv8-TFLite-Python/main.py
View file

@ -215,7 +215,7 @@ if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
detector = YOLOv8TFLite(args.model, args.conf, args.iou, args.metadata) detector = YOLOv8TFLite(args.model, args.conf, args.iou, args.metadata)
result = detector.detect(str(ASSETS / "bus.jpg"))[..., ::-1] result = detector.detect(str(ASSETS / "bus.jpg"))
cv2.imshow("Output", result) cv2.imshow("Output", result)
cv2.waitKey(0) cv2.waitKey(0)