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

Add OBB benchmarks to CI (#7777)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-01-23 19:40:41 +01:00 • committed by GitHub
parent 1152a06cbc
commit 3c1170769a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
ultralytics/utils/benchmarks.py
View file

@ -83,8 +83,12 @@ def benchmark(
for i, (name, format, suffix, cpu, gpu) in export_formats().iterrows(): # index, (name, format, suffix, CPU, GPU)
emoji, filename = "❌", None # export defaults
try:
assert i != 9 or LINUX, "Edge TPU export only supported on Linux"
if i in {5, 10}: # CoreML and TF.js
# Checks
if i == 9:
assert LINUX, "Edge TPU export only supported on Linux"
elif i == 7:
assert model.task != "obb", "TensorFlow GraphDef not supported for OBB task"
elif i in {5, 10}: # CoreML and TF.js
assert MACOS or LINUX, "export only supported on macOS and Linux"
if "cpu" in device.type:
assert cpu, "inference not supported on CPU"