Fix heatmaps width-height order bug (#7184)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
WangQvQ 2023-12-29 05:36:07 +08:00 committed by GitHub
parent 7499d72632
commit 79f588fa55
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -183,8 +183,8 @@ A heatmap generated with [Ultralytics YOLOv8](https://github.com/ultralytics/ult
# Heatmap Init
heatmap_obj = heatmap.Heatmap()
heatmap_obj.set_args(colormap=cv2.COLORMAP_PARULA ,
imw=im0.shape[1], # should same as im0 width
imh=im0.shape[0], # should same as im0 height
imw=im0.shape[0], # should same as im0 width
imh=im0.shape[1], # should same as im0 height
view_img=True,
shape="circle")