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

ultralytics 8.0.219 new save_frames=False predict arg (#6396)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Jason Sohn 2023-11-27 18:36:24 +09:00 • committed by GitHub
parent fdcf0dd4fd
commit 4096b261fc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 24 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/test_python.py
View file

@ -154,9 +154,10 @@ def test_track_stream():
"""
import yaml
video_url = 'https://ultralytics.com/assets/decelera_portrait_min.mov'
model = YOLO(MODEL)
model.track('https://ultralytics.com/assets/decelera_portrait_min.mov', imgsz=160, tracker='bytetrack.yaml')
model.track('https://ultralytics.com/assets/decelera_portrait_min.mov', imgsz=160, tracker='botsort.yaml')
model.track(video_url, imgsz=160, tracker='bytetrack.yaml')
model.track(video_url, imgsz=160, tracker='botsort.yaml', save_frames=True) # test frame saving also
# Test Global Motion Compensation (GMC) methods
for gmc in 'orb', 'sift', 'ecc':
@ -166,7 +167,7 @@ def test_track_stream():
data['gmc_method'] = gmc
with open(tracker, 'w', encoding='utf-8') as f:
yaml.safe_dump(data, f)
model.track('https://ultralytics.com/assets/decelera_portrait_min.mov', imgsz=160, tracker=tracker)
model.track(video_url, imgsz=160, tracker=tracker)
def test_val():