ultralytics / yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
50.03k stars 16.17k forks source link

Edge TPU Compilation #6513

Closed achelm15 closed 2 years ago

achelm15 commented 2 years ago

Search before asking

Question

I have been messing around with customizing the number of CSPBottleNecks layers in the C3 modules in YOLOv5 in order to try to gain some of the performance of the small version while maintaining the size of the nano version. I tried to take the original YOLOv5n and change only the C3 modules in layers 4 and 6 to the number they have in YOLOv5s. My goal is to put this on a Google Coral Devboard so I attempted to put use the Google Edge TPU Compiler, however it always ends in an error:

Edge TPU Compiler version 16.0.384591198 Started a compilation timeout timer of 180 seconds. Compilation child process completed within timeout period. Compilation failed!

Any ideas on how to fix this? Or a more appropriate place to ask this question?

Update: I have found that this occurs with the vanilla version of the YOLOv5s model as well.

Additional

No response

github-actions[bot] commented 2 years ago

👋 Hello @achelm15, thank you for your interest in YOLOv5 🚀! Please visit our ⭐️ Tutorials to get started, where you can find quickstart guides for simple tasks like Custom Data Training all the way to advanced concepts like Hyperparameter Evolution.

If this is a 🐛 Bug Report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom training ❓ Question, please provide as much information as possible, including dataset images, training logs, screenshots, and a public link to online W&B logging if available.

For business inquiries or professional support requests please visit https://ultralytics.com or email support@ultralytics.com.

Requirements

Python>=3.7.0 with all requirements.txt installed including PyTorch>=1.7. To get started:

git clone https://github.com/ultralytics/yolov5  # clone
cd yolov5
pip install -r requirements.txt  # install

Environments

YOLOv5 may be run in any of the following up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python and PyTorch preinstalled):

Status

CI CPU testing

If this badge is green, all YOLOv5 GitHub Actions Continuous Integration (CI) tests are currently passing. CI tests verify correct operation of YOLOv5 training (train.py), validation (val.py), inference (detect.py) and export (export.py) on MacOS, Windows, and Ubuntu every 24 hours and on every commit.

glenn-jocher commented 2 years ago

@achelm15 code and model customizations are outside the scope of our official support but perhaps a community member has some ideas.