`ultralytics 8.1.18` add cmake for building onnxsim on aarch64 #8395

lakshanthad · 2024-02-22T20:44:15Z

onnxsim does not have pre-built binaries for aarch64.
https://pypi.org/project/onnxsim/#files

When exporting to tflite and exporting to onnx with simplify on aarch64 devices, it will try to download the tar.gz (20.1MB) file and compile onnxsim on the device. However, this needs cmake for the build process. All the other architectures have pre-built binaries and have no problem.

So I have included cmake inside onnx export with simplify and tflite exports. The reason I included cmake separately without combining it with other packages is that if cmake is included together with onnxsim inside check_requirements(), it will have the following error:

pi@raspberrypi:~ $ yolo export model=yolov8n.pt format=onnx simplify
Ultralytics YOLOv8.1.16 🚀 Python-3.11.2 torch-2.2.0 CPU (Cortex-A76)
YOLOv8n summary (fused): 168 layers, 3151904 parameters, 0 gradients, 8.7 GFLOPs

PyTorch: starting from 'yolov8n.pt' with input shape (1, 3, 640, 640) BCHW and output shape(s) (1, 84, 8400) (6.2 MB)
requirements: Ultralytics requirements ['cmake', 'onnxsim>=0.4.33'] not found, attempting AutoUpdate...
  error: subprocess-exited-with-error

  × python setup.py egg_info did not run successfully.
  │ exit code: 1
  ╰─> [8 lines of output]
      fatal: not a git repository (or any of the parent directories): .git
      fatal: not a git repository (or any of the parent directories): .git
      Traceback (most recent call last):
        File "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 34, in <module>
        File "/tmp/pip-install-6er09w5u/onnxsim_9b0aac316a144e64a73932b33fa051c1/setup.py", line 67, in <module>
          assert CMAKE, 'Could not find "cmake" executable!'
      AssertionError: Could not find "cmake" executable!
      [end of output]

  note: This error originates from a subprocess, and is likely not a problem with pip.
error: metadata-generation-failed

× Encountered error while generating package metadata.
╰─> See above for output.

note: This is an issue with the package mentioned above, not pip.
hint: See above for details.
requirements: ❌ Command 'pip install --no-cache "cmake" "onnxsim>=0.4.33" ' returned non-zero exit status 1.

So cmake need to be installed separately at first.

I have read the CLA Document and I hereby sign the CLA

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

Improved ARM64 Dockerfile and package requirements management.

📊 Key Changes

Removed cmake from Dockerfile for ARM64 builds, optimizing the installation layer.
Updated the Ultralytics package version from 8.1.17 to 8.1.18.
Added conditional checks to install cmake if exporting to ONNX on an ARM64 system.
Adjusted TensorFlow package requirement logic to handle ARM64 architecture and restricted versioning for non-ARM64 systems.
Restricted flatbuffers version to below 100 to avoid potential future conflicts.

🎯 Purpose & Impact

Purpose: These changes streamline the Docker build process for ARM64 and ensure compatibility and stability when exporting models using different platforms.
Impact: Users benefit from a smoother experience with Docker on ARM64, and developers gain clarity on package dependencies during model exporting, particularly with TensorFlow and ONNX. This may also enhance overall performance and maintainability of the Ultralytics codebase. 🚀

Currently tensorflow PyPI includes pre-build binaries for all architectures. Therefore, there is no need to specify different architectures. https://pypi.org/project/tensorflow/2.15.0/#files

onnxsim does not have pre-built binaries for aarch64. https://pypi.org/project/onnxsim/#files When exporting to tflite and exporting to onnx with simplify on aarch64 devices, it will try to download the tar.gz (20.1MB) file and compile onnxsim on the device. However this needs 'cmake' for the build process.

lakshanthad · 2024-02-22T20:46:07Z

@glenn-jocher FYI

codecov · 2024-02-22T20:47:07Z

Codecov Report

Attention: Patch coverage is 28.57143% with 5 lines in your changes are missing coverage. Please review.

Project coverage is 75.61%. Comparing base (8c28e0c) to head (06f2365).

❗ Current head 06f2365 differs from pull request most recent head cfce540. Consider uploading reports for the commit cfce540 to get more accurate results

Files	Patch %	Lines
ultralytics/engine/exporter.py	28.57%	5 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #8395      +/-   ##
==========================================
- Coverage   75.83%   75.61%   -0.23%     
==========================================
  Files         117      117              
  Lines       14673    14679       +6     
==========================================
- Hits        11127    11099      -28     
- Misses       3546     3580      +34

Flag	Coverage Δ
Benchmarks	`36.17% <28.57%> (-0.17%)`	⬇️
GPU	`39.02% <0.00%> (-0.03%)`	⬇️
Tests	`70.69% <14.28%> (-0.19%)`	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

glenn-jocher · 2024-02-22T21:09:19Z

Maybe do this?

suffix = '-macos' if MACOS else '-aarch64' if ARM64 else '' if cuda else '-cpu'
version = '' if ARM64 else '<=12.13.1'
check_requirements(f"tensorflow{suffix}{version}")

glenn-jocher · 2024-02-22T21:16:27Z

@lakshanthad ok I've closed #8345 in favor of this PR. Updated the code slightly and waiting on CI now.

lakshanthad · 2024-02-22T21:23:40Z

Maybe do this?

suffix = '-macos' if MACOS else '-aarch64' if ARM64 else '' if cuda else '-cpu'
version = '' if ARM64 else '<=12.13.1'
check_requirements(f"tensorflow{suffix}{version}")

@glenn-jocher
typo there. should be 2.13.1. its fixed now. thanks for the formatting.

glenn-jocher · 2024-02-22T21:33:09Z

Maybe do this?
suffix = '-macos' if MACOS else '-aarch64' if ARM64 else '' if cuda else '-cpu'
version = '' if ARM64 else '<=12.13.1'
check_requirements(f"tensorflow{suffix}{version}")
@glenn-jocher typo there. should be 2.13.1. its fixed now. thanks for the formatting.

oh yeah haha, good catch!

glenn-jocher · 2024-02-22T22:43:32Z

@lakshanthad PR merged with patch version bump, nice work on this!

lakshanthad · 2024-02-22T22:56:51Z

Great! @glenn-jocher Thank you!

@Retry

* Explorer API video https://youtu.be/3VryynorQeo (ultralytics#7838) * Add HUB-SDK Docs reference section (ultralytics#7781) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com> * Link checks SSL insecure robustness (ultralytics#7853) * Add new @Retry() decorator (ultralytics#7854) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add TensorRT Docs Integrations Page (ultralytics#7855) * Cleanup Docs languages (ultralytics#7865) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add millimeters in `solutions/distance_caculation.py` + `object-cropping.md` visuals (ultralytics#7860) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.7` `USER_CONFIG_DIR` Explorer ops (ultralytics#7861) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Ultralytics Actions with OpenAI GPT-4 PR Summary (ultralytics#7867) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Bump slackapi/slack-github-action from 1.24.0 to 1.25.0 in /.github/workflows (ultralytics#7871) Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> * Add visuals in `guides/distance-calculation.md` (ultralytics#7876) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * TensorRT FP16 export with `dynamic=True` (ultralytics#7870) * Add https://youtu.be/3VryynorQeo to README and fix `converter.py` Docs (ultralytics#7883) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * `ultralytics 8.1.8` new `model.save('filename.pt')` method (ultralytics#7886) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Compress explorer image in `dataset/index.md` + description added for new `segmentation` datasets (ultralytics#7901) * Fix OpenVINO links (ultralytics#7930) * Add `integrations/gradio` Docs page (ultralytics#7935) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: WangQvQ <1579093407@qq.com> Co-authored-by: Martin Pl <martin-plank@gmx.de> Co-authored-by: Mactarvish <Mactarvish@users.noreply.github.com> * Add https://youtu.be/96NkhsV-W1U to docs and `heatmaps` fix (ultralytics#7944) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Segment TensorRT `batch=3` fix (ultralytics#7952) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.9` replace `.size(0)` with `.shape[0]` (ultralytics#7957) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add YouTube iframe `loading="lazy"` (ultralytics#8001) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Mkdocs updates (ultralytics#8008) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Tracker: clean up GMC (ultralytics#8003) * Bump nick-invision/retry from 2 to 3 in /.github/workflows (ultralytics#8023) Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> * Bump codecov/codecov-action from 3 to 4 in /.github/workflows (ultralytics#8022) Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> * Fix TFLite INT8 for OBB (ultralytics#7989) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add CoreML Docs Integrations Page (ultralytics#8063) * Add Roboflow 100 Dataset Docs Page (ultralytics#8065) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.10` MLFlow, OBB, TFLite and INT8 fixes (ultralytics#8016) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add https://youtu.be/R42s2zFtNIY to `hub/datasets.md` and CoreML image fix (ultralytics#8085) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update Roboflow segmentation Docs pages (ultralytics#8094) * FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime (ultralytics#8104) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.11` expand OpenVINO INT8 ops for improved mAP (ultralytics#7516) Co-authored-by: AdamP <7806910+adamp87@users.noreply.github.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Add https://youtu.be/q7LwPoM7tSQ to `guides/yolo-performance-metrics.md` (ultralytics#8114) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Compress docs images (ultralytics#8124) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Updated SQL Query Plot Code (ultralytics#8120) Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Elaborate Train and Predict mode docs arguments (ultralytics#8137) Signed-off-by: Skillnoob <78843978+Skillnoob@users.noreply.github.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Skillnoob <78843978+Skillnoob@users.noreply.github.com> Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com> * Move Google Drive to slow tests (ultralytics#8163) * Improve Coral Edge TPU guide (ultralytics#8160) Signed-off-by: Skillnoob <78843978+Skillnoob@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Try Apple M1 runners for Tests and Benchmarks CI (ultralytics#8162) * Retry once for Docker CI failures (ultralytics#8164) * `ultralytics 8.1.12` new `Results.show()` and `Results.save()` (ultralytics#8166) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Fix solution functions description + `integrations/openvino.md` cover image update (ultralytics#8170) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Pin `duckdb<=0.9.2` to avoid 0.10.0 errors (ultralytics#8181) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * New HUB Cloud Training docs page (ultralytics#8174) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.13` fix `yolo train time=` bug (ultralytics#8179) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Docs updates and improvements (ultralytics#8183) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Priytosh Tripathi <priytosh.revolution@live.com> * Fix xyxyxyxyn calculation in obb result (ultralytics#8188) * `ultralytics 8.1.14` new YOLOv8-World models (ultralytics#8054) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix PIL `show()` and `save()` (ultralytics#8202) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Remove OpenVINO INT8 RTDETRDecoder ops (ultralytics#8219) * Fix xyxyxyxyn calculation, swap axis (ultralytics#8249) * Improve Docs Modes tables (ultralytics#8266) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.15` add Python 3.12 compatibility (ultralytics#8210) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Johnny <johnnynuca14@gmail.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Expand `Model` method type hinting (ultralytics#8279) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.16` OBB ConfusionMatrix support (ultralytics#8299) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Move Explorer tests to scheduled CI (ultralytics#8305) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Remove JAX constraints in Exporter (ultralytics#8309) * `Model` typehints Docker fix (ultralytics#8306) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix an issue with flatbuffer version for Raspberry Pi (ultralytics#8311) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update Dockerfile for Raspberry Pi Support (ultralytics#8123) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Avoid creating `-hub` dir without processing images (ultralytics#8334) * Added check of parameter type before setting `required_grad=True` for frozen layers (ultralytics#8318) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add `estimate_speed(region_color)` parameter (ultralytics#8285) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add simple-utilities.md Docs page (ultralytics#8269) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update `mkdocs-ultralytics-plugin>=0.0.44` (ultralytics#8347) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.17` fix `ClassificationDataset` caching (ultralytics#8358) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add missing single-line docstrings (ultralytics#8362) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Johannes Kaisinger <johannes.kaisinger@gmail.com> * `ultralytics 8.1.18` add cmake for building onnxsim on aarch64 (ultralytics#8395) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Allow Annotator PIL.Image inputs (ultralytics#8397) * Add Non-Maximum Suppression (NMS) `inplace` flag (ultralytics#8368) * Compare `plt.get_backend()` in lowercase (ultralytics#8409) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Improve Docs arguments tables (ultralytics#8415) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add plot_images `conf_thresh` parameter (ultralytics#8446) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add YOLOv9 Docs page (ultralytics#8478) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Rename `model_id` to `model.id` (ultralytics#8447) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.19` PNNX `aarch64` linux fix (ultralytics#8480) Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com> Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com> * Add TorchScript Docs Integrations Page (ultralytics#8501) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Do RTDETR file suffix check using pathlib instead of string manipulations (ultralytics#8525) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Auto-format by https://ultralytics.com/actions * Add TFLite Docs Integrations Page (ultralytics#8522) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix undefined ‘im_array’ bug in predict.md (ultralytics#8565) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: fang_chenfang <1217690899@qq.com> * Add https://youtu.be/ie3vLUDNYZo and other YT videos in Docs (ultralytics#8551) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Hold failed upload metrics and upload with next upload metrics (ultralytics#8513) Co-authored-by: hassaanfarooq01 <hassaanfarooq01@gmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.20` add YOLOv8x-World support (ultralytics#8539) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix device counting method to account for double-digit device IDs (ultralytics#8502) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Remove unused variables (ultralytics#8511) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add NCNN Docs Integrations Page (ultralytics#8562) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Optimize function calls to method calls (ultralytics#8507) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Replacing Images for torchscript.md (ultralytics#8596) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix cmake requirements for ARM64 (ultralytics#8589) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add https://youtu.be/rCggzXRRSRo to Speed Estimation Docs (ultralytics#8577) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add KaTeX support + updates to YOLOv9 model page (ultralytics#8531) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Replace `onnx2tf` usage from CLI to Python (ultralytics#8429) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update IoU capitalization (ultralytics#8604) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Dean Mark <2552482+deanmark@users.noreply.github.com> * Update TFLite Docs images (ultralytics#8605) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add `distance calculation` feature in `vision-eye` (ultralytics#8616) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Muhammad Rizwan Munawar <chr043416@gmail.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * `ultralytics 8.1.21` Add YOLOv8-World-v2 models (ultralytics#8580) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * `ultralytics 8.1.22` HUB model `pathlib` fix (ultralytics#8621) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.23` add YOLOv9-C and E models (ultralytics#8571) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add settings `runs_dir` == `datasets_dir` warning (ultralytics#8653) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add Ultralytics HUB Cloud Training banner to Docs (ultralytics#8656) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update CITATION.cff with cffinit corrections (ultralytics#8658) * Update Colab notebook for YOLOv8.1 (ultralytics#8660) * `ultralytics 8.1.24` new OpenVINO 2023.3 export updates (ultralytics#8417) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * OBB: Fix when training on zip data (ultralytics#8680) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Object Counter improvements (ultralytics#8648) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Clarify triton server boolean value (ultralytics#8532) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add `model.eval()` in TensorBoad graph visualization to avoid BN stats changes (ultralytics#8629) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add domain checks to CI docs page (ultralytics#8714) * `ultralytics 8.1.25` OpenVINO `LATENCY` and `THROUGHPUT` modes (ultralytics#8058) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Adrian Boguszewski <adekboguszewski@gmail.com> * Add https://youtu.be/OpWpBI35A5Y to inference-api Docs (ultralytics#8717) * OpenVino 2024.0.0 (ultralytics#8726) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix tfjs exports for ARM64 Dockerfile (ultralytics#8766) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Modify tuner best.pt logic to train first (ultralytics#8792) * Update CI.md with CI links (ultralytics#8786) * Add https://youtu.be/ZzUSXQkLbNw to Integration Docs (ultralytics#8793) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.25` fix `**kwargs: (dict)` warnings (ultralytics#8815) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add docs guide terminal images (ultralytics#8819) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `RTDETRDetectionModel` TorchScript, ONNX Predict and Val support (ultralytics#8818) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * New `MLFLOW_KEEP_RUN_ACTIVE` flag for enhanced MLflow run management (ultralytics#8808) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.26` `LoadImagesAndVideos` batched inference (ultralytics#8817) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update `fraction` arg to employ a random selection (ultralytics#8234) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Batch inference fixes (ultralytics#8854) * Integrate OpenVINO `CUMULATIVE_THROUGHPUT` mode batched inference (ultralytics#8834) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add PaddlePaddle Docs Integrations Page (ultralytics#8858) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.27` batched tracking fixes (ultralytics#8842) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Faster batch `same_shapes` (ultralytics#8851) * More robust PNNX downloads (ultralytics#8866) * Fix `test_mlflow_keep_run_active()` (ultralytics#8868) * Add polygon regions drawing support in `object-counting.md` and minor docs update (ultralytics#8885) * Add TFLite Edge TPU Docs Integrations Page (ultralytics#8900) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update Docs README (ultralytics#8919) * Fix `names` bug when exporting YOLOv8-World to ONNX (ultralytics#8941) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * New test for labels and crops (ultralytics#8861) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Optimize TFJS export on ARM64 (ultralytics#8946) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Auto-format by https://ultralytics.com/actions * `ultralytics 8.1.28` avoid * ops on bool Tensors for RT-DETR OpenVINO export (ultralytics#8937) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add https://youtu.be/YDXKa1EljmU and https://youtu.be/5BO0Il_YYAg to Docs (ultralytics#8958) * `ultralytics 8.1.29` improved disk space checking on correct path (ultralytics#8977) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com> * Add OpenVINO Latency vs Throughput modes Docs page (ultralytics#9001) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add `brain-tumor` dataset docs page (ultralytics#9041) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix Neural Magic links (ultralytics#9144) * Updating a typo on paddlepaddle.md (ultralytics#9108) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.30` add advanced HUB train arguments (ultralytics#9110) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * fix yolo classify model loading error (ultralytics#9196) * Add TF GraphDef Docs Integrations Page (ultralytics#9203) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.31` NCNN and CLIP updates (ultralytics#9235) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix TypeError in check_cfg() (ultralytics#9245) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add https://youtu.be/aeAX6vWpfR0 to docs & `african-wildlife` dataset docs page (ultralytics#9102) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add TF SavedModel Docs Integrations Page (ultralytics#9162) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Limit `ray<2.9.3` due to bug in `ray==2.10.0` (ultralytics#9254) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.32` fix CLIP backwards compatibility (ultralytics#9253) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Conda cv2 ImportError libEGL.so.1 fix (ultralytics#9255) * Fix `IS_PYTHON_3_12` bug (ultralytics#9258) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> * Reverse IoU threshold for NMS docs descriptions (ultralytics#9151) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add `bgr` hyperparameter (ultralytics#9139) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update pyproject.toml `[tool.pytest.ini_options]` (ultralytics#9259) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix OpenVINO missing batch-size bug (ultralytics#9264) * `ultralytics 8.1.33` fix HUB model checks (ultralytics#9153) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update Discord and Contributing Guide URLs (ultralytics#9270) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Add https://youtu.be/uDrn9QZJ2lk and `object_counting.py`, `ai_gym.py` updates (ultralytics#9265) * `ultralytics 8.1.34` Inference API robust imgsz checks (ultralytics#9274) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Bump contributor-assistant/github-action from 2.3.1 to 2.3.2 in /.github/workflows (ultralytics#9279) Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> * Add https://youtu.be/u3pLlgzUeV8 to docs (ultralytics#9367) * Fix typo in YOLOv8-Libtorch-CPP-Inference (ultralytics#9330) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Simplify metrics calculation (ultralytics#9338) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Fix typo in example readme (ultralytics#9306) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Specify YOLODataset task arg during INT8 calibration (ultralytics#9309) Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> * `ultralytics 8.1.35` simplify network modules (ultralytics#9321) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> --------- Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: dependabot[bot] <support@github.com> Signed-off-by: Skillnoob <78843978+Skillnoob@users.noreply.github.com> Co-authored-by: Muhammad Rizwan Munawar <chr043416@gmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com> Co-authored-by: Abirami Vina <abirami.vina@gmail.com> Co-authored-by: Paula Derrenger <107626595+pderrenger@users.noreply.github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com> Co-authored-by: Muhammad Rizwan Munawar <muhammadrizwanmunawar123@gmail.com> Co-authored-by: WangQvQ <1579093407@qq.com> Co-authored-by: Martin Pl <martin-plank@gmx.de> Co-authored-by: Mactarvish <Mactarvish@users.noreply.github.com> Co-authored-by: DieterTimmers <46112022+DieterTimmers@users.noreply.github.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: AdamP <7806910+adamp87@users.noreply.github.com> Co-authored-by: Youho99 <44434482+Youho99@users.noreply.github.com> Co-authored-by: Ankan Ghosh <79740115+0xSynapse@users.noreply.github.com> Co-authored-by: Skillnoob <78843978+Skillnoob@users.noreply.github.com> Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com> Co-authored-by: Priytosh Tripathi <priytosh.revolution@live.com> Co-authored-by: Johannes Kaisinger <johannes.kaisinger@gmail.com> Co-authored-by: Johnny <johnnynuca14@gmail.com> Co-authored-by: Lakshantha Dissanayake <lakshanthad@yahoo.com> Co-authored-by: Alexander Suslov <alexander.suslov@intel.com> Co-authored-by: Chi <iamchi@skiff.com> Co-authored-by: AlainSchoebi <44315825+AlainSchoebi@users.noreply.github.com> Co-authored-by: Dean Mark <2552482+deanmark@users.noreply.github.com> Co-authored-by: Yifei Liu <71677542+kaikai23@users.noreply.github.com> Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com> Co-authored-by: Robin Brown <birdcolour@users.noreply.github.com> Co-authored-by: fang_chenfang <1217690899@qq.com> Co-authored-by: Hassaan Farooq <103611273+hassaanfarooq01@users.noreply.github.com> Co-authored-by: hassaanfarooq01 <hassaanfarooq01@gmail.com> Co-authored-by: Omar Duhaiby <3omarz@gmail.com> Co-authored-by: Andrew <ykk8j4yq@duck.com> Co-authored-by: Sergiu Waxmann <47978446+sergiuwaxmann@users.noreply.github.com> Co-authored-by: Adrian Boguszewski <adekboguszewski@gmail.com> Co-authored-by: Henry <contact@henrynavarro.org> Co-authored-by: bobyard-com <154289614+bobyard-com@users.noreply.github.com> Co-authored-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Co-authored-by: Finlay Morrison <57669260+finlaymorrison@users.noreply.github.com> Co-authored-by: Massimiliano Riva <48362794+massimiliano96@users.noreply.github.com> Co-authored-by: Jamjamjon <51357717+jamjamjon@users.noreply.github.com> Co-authored-by: Kalen Michael <kalenmike@gmail.com> Co-authored-by: Mo Li <82895469+DseidLi@users.noreply.github.com> Co-authored-by: zhaoruibing <zhaoruibing@gmail.com> Co-authored-by: Dzmitry Plashchynski <plashchynski@gmail.com> Co-authored-by: higor-melo <115033428+higor-melo@users.noreply.github.com> Co-authored-by: Eric Hanson <5846501+ericphanson@users.noreply.github.com> Co-authored-by: sifan-intel <109498696+sifan-intel@users.noreply.github.com>

…lytics#8395) Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>

lakshanthad and others added 6 commits February 20, 2024 13:30

Keep tensorflow common to all architectures

71912fc

Currently tensorflow PyPI includes pre-build binaries for all architectures. Therefore, there is no need to specify different architectures. https://pypi.org/project/tensorflow/2.15.0/#files

Merge branch 'main' into main

a720d1d

Merge branch 'ultralytics:main' into main

dbc1621

Fix tensorflow architectures

06d46dd

Auto-format by https://ultralytics.com/actions

72b3592

Update exporter.py

d73cfd1

glenn-jocher and others added 2 commits February 22, 2024 22:13

Update exporter.py

7107772

Auto-format by https://ultralytics.com/actions

0bc30ea

glenn-jocher mentioned this pull request Feb 22, 2024

Update tensorflow versions according to architecture #8345

Closed

Update exporter.py

ea3c080

Merge branch 'main' into tflite

1028c95

Update exporter.py

06f2365

Update __init__.py

cfce540

glenn-jocher changed the title ~~Add 'cmake' for building onnxsim on aarch64~~ ultralytics 8.1.18 add cmake for building onnxsim on aarch64 Feb 22, 2024

glenn-jocher merged commit 0572b29 into ultralytics:main Feb 22, 2024
10 checks passed

lakshanthad mentioned this pull request May 31, 2024

cmake ARM64 fix #13260

Closed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

`ultralytics 8.1.18` add cmake for building onnxsim on aarch64 #8395

`ultralytics 8.1.18` add cmake for building onnxsim on aarch64 #8395

lakshanthad commented Feb 22, 2024 •

edited by github-actions bot

Loading

lakshanthad commented Feb 22, 2024

codecov bot commented Feb 22, 2024 •

edited

Loading

glenn-jocher commented Feb 22, 2024

glenn-jocher commented Feb 22, 2024

lakshanthad commented Feb 22, 2024 •

edited

Loading

glenn-jocher commented Feb 22, 2024

glenn-jocher commented Feb 22, 2024

lakshanthad commented Feb 22, 2024

ultralytics 8.1.18 add cmake for building onnxsim on aarch64 #8395

ultralytics 8.1.18 add cmake for building onnxsim on aarch64 #8395

Conversation

lakshanthad commented Feb 22, 2024 • edited by github-actions bot Loading

🛠️ PR Summary

🌟 Summary

📊 Key Changes

🎯 Purpose & Impact

lakshanthad commented Feb 22, 2024

codecov bot commented Feb 22, 2024 • edited Loading

Codecov Report

glenn-jocher commented Feb 22, 2024

glenn-jocher commented Feb 22, 2024

lakshanthad commented Feb 22, 2024 • edited Loading

glenn-jocher commented Feb 22, 2024

glenn-jocher commented Feb 22, 2024

lakshanthad commented Feb 22, 2024

`ultralytics 8.1.18` add cmake for building onnxsim on aarch64 #8395

`ultralytics 8.1.18` add cmake for building onnxsim on aarch64 #8395

lakshanthad commented Feb 22, 2024 •

edited by github-actions bot

Loading

codecov bot commented Feb 22, 2024 •

edited

Loading

lakshanthad commented Feb 22, 2024 •

edited

Loading