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

Fix missing data.yaml error on int8 export (#19128)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Mohammed Yasin 2025-02-07 20:07:32 +08:00 • committed by GitHub
parent 64c835dc88
commit 580b4766c5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ultralytics/utils/benchmarks.py
View file

@ -152,7 +152,9 @@ def benchmark(
filename = model.pt_path or model.ckpt_path or model.model_name filename = model.pt_path or model.ckpt_path or model.model_name
exported_model = model # PyTorch format exported_model = model # PyTorch format
else: else:
filename = model.export(imgsz=imgsz, format=format, half=half, int8=int8, device=device, verbose=False) filename = model.export(
imgsz=imgsz, format=format, half=half, int8=int8, data=data, device=device, verbose=False
)
exported_model = YOLO(filename, task=model.task) exported_model = YOLO(filename, task=model.task)
assert suffix in str(filename), "export failed" assert suffix in str(filename), "export failed"
emoji = "❎" # indicates export succeeded emoji = "❎" # indicates export succeeded
@ -163,7 +165,7 @@ def benchmark(
assert i != 5 or platform.system() == "Darwin", "inference only supported on macOS>=10.13" # CoreML assert i != 5 or platform.system() == "Darwin", "inference only supported on macOS>=10.13" # CoreML
if i in {13}: if i in {13}:
assert not is_end2end, "End-to-end torch.topk operation is not supported for NCNN prediction yet" assert not is_end2end, "End-to-end torch.topk operation is not supported for NCNN prediction yet"
exported_model.predict(ASSETS / "bus.jpg", imgsz=imgsz, device=device, half=half, data=data, verbose=False) exported_model.predict(ASSETS / "bus.jpg", imgsz=imgsz, device=device, half=half, verbose=False)
# Validate # Validate
results = exported_model.val( results = exported_model.val(