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

Failing CUDA tests fixes (#4682)

Browse source
This commit is contained in:
Glenn Jocher 2023-09-01 13:29:08 +02:00 • committed by GitHub
parent 263bfd1e93
commit 2bc6e647c7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/test_cuda.py
View file

@ -71,7 +71,7 @@ def test_predict_sam():
predictor = SAMPredictor(overrides=overrides)
# Set image
predictor.set_image('ultralytics/assets/zidane.jpg') # set with image file
predictor.set_image(ASSETS / 'zidane.jpg') # set with image file
# predictor(bboxes=[439, 437, 524, 709])
# predictor(points=[900, 370], labels=[1])

6
tests/test_python.py
View file

@ -140,11 +140,11 @@ def test_track_stream():
# Test Global Motion Compensation (GMC) methods
for gmc in 'orb', 'sift', 'ecc':
with open(ROOT / 'cfg/trackers/botsort.yaml') as f:
with open(ROOT / 'cfg/trackers/botsort.yaml', encoding='utf-8') as f:
data = yaml.safe_load(f)
tracker = TMP / f'botsort-{gmc}.yaml'
data['gmc_method'] = gmc
with open(tracker, 'w') as f:
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)
@ -166,7 +166,7 @@ def test_train_pretrained():
def test_export_torchscript():
f = YOLO(MODEL).export(format='torchscript', optimize=True)
f = YOLO(MODEL).export(format='torchscript', optimize=False)
YOLO(f)(SOURCE) # exported model inference