ultralytics 8.0.230 TensorRT export hang fix (#7180)
Co-authored-by: 曾逸夫(Zeng Yifu) <41098760+Zengyf-CVer@users.noreply.github.com> Co-authored-by: CV & LLM & AIGC er <wenvoi@163.com> Co-authored-by: Aaron <42322215+aaronllowe@users.noreply.github.com> Co-authored-by: crypthusiast0 <42322215+crypthusiast0@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
parent
bc5b528ca7
commit
8d2a32db77
6 changed files with 15 additions and 7 deletions
|
|
@ -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=cap.get(4), # should same as cap height
|
||||
imh=cap.get(3), # should same as cap width
|
||||
imw=im0.shape[1], # should same as im0 width
|
||||
imh=im0.shape[0], # should same as im0 height
|
||||
view_img=True,
|
||||
shape="circle")
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue