mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 21:44:22 +08:00
Fix heatmaps width-height order bug (#7184)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
7499d72632
commit
79f588fa55
@ -183,8 +183,8 @@ A heatmap generated with [Ultralytics YOLOv8](https://github.com/ultralytics/ult
|
|||||||
# Heatmap Init
|
# Heatmap Init
|
||||||
heatmap_obj = heatmap.Heatmap()
|
heatmap_obj = heatmap.Heatmap()
|
||||||
heatmap_obj.set_args(colormap=cv2.COLORMAP_PARULA ,
|
heatmap_obj.set_args(colormap=cv2.COLORMAP_PARULA ,
|
||||||
imw=im0.shape[1], # should same as im0 width
|
imw=im0.shape[0], # should same as im0 width
|
||||||
imh=im0.shape[0], # should same as im0 height
|
imh=im0.shape[1], # should same as im0 height
|
||||||
view_img=True,
|
view_img=True,
|
||||||
shape="circle")
|
shape="circle")
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user