mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
yolov5/requirements.txt

44 lines
1.1 KiB
Plaintext
Raw Normal View History

Update requirements.txt with Usage example (#7966) Update requirements.txt
2022-05-25 07:40:17 +08:00
# YOLOv5 requirements
# Usage: pip install -r requirements.txt
update requirements.txt
2020-08-15 01:14:48 +08:00
Add TensorFlow formats to `export.py` (#4479) * Initial commit * Remove unused export_torchscript return * ROOT variable * Add prefix to fcn arg * fix ROOT * check_yaml into run() * interim fixes * imgsz=(320, 320) * Hardcode tf_raw_resize False * Finish opt elimination * Update representative_dataset_gen() * Update export.py with TF methods * SiLU and GraphDef fixes * file_size() directory handling feature * export fixes * add lambda: to representative_dataset * Detect training False default * Fuse false for TF models * Embed agnostic NMS arguments * Remove lambda * TensorFlow.js export success * Add pb to Usage * Add *_tfjs_model/ to ignore files * prepend YOLOv5 to function headers * Remove end --- comments * parameterize tfjs export pb file * update run() data default /ROOT * update --include help * update imports * return ct_model * Consolidate TFLite export * pb prerequisite to tfjs * TF modules CamelCase * Remove exports from tf.py and cleanup * pass agnostic NMS arguments * CI * CI * ignore *_web_model/ * Add tensorflow to CI dependencies * CI tensorflow-cpu * Update requirements.txt * Remove tensorflow check_requirement * CI coreml tfjs * export only onnx torchscript * reorder exports torchscript first
2021-09-12 21:52:24 +08:00
# Base ----------------------------------------
update requirements.txt
2020-08-02 05:00:14 +08:00
matplotlib>=3.2.2
update requirements.txt
2020-07-17 04:37:58 +08:00
numpy>=1.18.5
Reduce `opencv-python>=4.1.1` for Jetson Nano (#7645) May help https://github.com/ultralytics/yolov5/issues/7631#issuecomment-1113007451
2022-04-30 02:50:30 +08:00
opencv-python>=4.1.1
`PIL.ImageDraw.text(anchor=...)` removal, reduce to `>=7.1.2` (#4842) * Unpin Pillow * Update requirements.txt * Update plots.py
2021-09-17 19:02:37 +08:00
Pillow>=7.1.2
Docker pyYAML>=5.3.1 fix (#2031)
2021-01-25 03:54:15 +08:00
PyYAML>=5.3.1
Add `requests` to requirements.txt (#5112) * Requirements updated 1. Requests added to requirements.txt. That might not be included in all docker base images, adding it to the requirements is safer. 2. Added a minimum version to Pandas. It's a good practice to have versions for all dependencies. * Sort alphabetically * Update requirements.txt Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-10 06:51:01 +08:00
requests>=2.23.0
Update `tqdm>=4.64.0` and `thop>=0.1.1` (#8596) * Update `tqdm>=4.64.0` and `thop>=0.1.1` * Update requirements.txt
2022-07-16 22:14:16 +08:00
scipy>=1.4.1
assert torch!=1.12.0 for DDP training (#8621) * assert torch!=1.12.0 for DDP training * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-07-18 21:05:58 +08:00
torch>=1.7.0
torchvision>=0.8.1
Update `tqdm>=4.64.0` and `thop>=0.1.1` (#8596) * Update `tqdm>=4.64.0` and `thop>=0.1.1` * Update requirements.txt
2022-07-16 22:14:16 +08:00
tqdm>=4.64.0
Comment `protobuf` in requirements.txt (#9207) The low package version is causing conflicts among other dependencies, commenting it causes no ill effects in CI so this should be fine. Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-29 23:13:55 +08:00
# protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012
initial commit
2020-05-30 08:04:54 +08:00
Add TensorFlow formats to `export.py` (#4479) * Initial commit * Remove unused export_torchscript return * ROOT variable * Add prefix to fcn arg * fix ROOT * check_yaml into run() * interim fixes * imgsz=(320, 320) * Hardcode tf_raw_resize False * Finish opt elimination * Update representative_dataset_gen() * Update export.py with TF methods * SiLU and GraphDef fixes * file_size() directory handling feature * export fixes * add lambda: to representative_dataset * Detect training False default * Fuse false for TF models * Embed agnostic NMS arguments * Remove lambda * TensorFlow.js export success * Add pb to Usage * Add *_tfjs_model/ to ignore files * prepend YOLOv5 to function headers * Remove end --- comments * parameterize tfjs export pb file * update run() data default /ROOT * update --include help * update imports * return ct_model * Consolidate TFLite export * pb prerequisite to tfjs * TF modules CamelCase * Remove exports from tf.py and cleanup * pass agnostic NMS arguments * CI * CI * ignore *_web_model/ * Add tensorflow to CI dependencies * CI tensorflow-cpu * Update requirements.txt * Remove tensorflow check_requirement * CI coreml tfjs * export only onnx torchscript * reorder exports torchscript first
2021-09-12 21:52:24 +08:00
# Logging -------------------------------------
Update tensorboard>=2.4.1 (#2576) * Update tensorboard>=2.4.1 Update tensorboard version to attempt to address https://github.com/ultralytics/yolov5/issues/2573 (tensorboard logging fail in Docker image). * cleanup
2021-03-24 08:29:00 +08:00
tensorboard>=2.4.1
Weights & Biases (W&B) Feature Addition (#1235) * Add wandb metric logging and bounding box debugging * Improve formatting, readability * Remove mutliple path for init, improve formatting * Add wandb params * Remove typecasting in bbox coordinates and reformat * Cleanup * add wandb to requirements.txt * minor updates to test.py * general reorg * reduce --log-imgs to 10 * clean wandb import * reverse wandb import assert * add except AssertionError to try import * move wandb init to all global ranks * replace print() with logger.info() * replace print() with logger.info() * move wandb.init() bug fix * project PosixPath to basename bug fix Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-11-01 03:16:35 +08:00
# wandb
ClearML experiment tracking integration (#8620) * Add titles to matplotlib plots * Add ClearML Experiment Tracking integration. * Add ClearML Data Version Management automatic download when requested * Add ClearML Hyperparameter Optimization * ClearML save period integration * Fix wandb breaking when used with ClearML dataset * Fix wandb breaking when used with ClearML resume and dataset * Add ClearML documentation * fixed small bug in clearml integration that misreports epoch number * Final ClearMl additions before refactor * Add correct epoch reporting * Add remote execution and autoscaling docs for ClearML integration * Added images to clearml integration docs * fixed logo alignment bug and added hpo screenshot clearml * Fixed small epoch number bug in clearml integration * Remove saved model flush clearml * Cleanup clearml readme section * Cleaned up clearml logger docstring * Remove resume readme section clearml * Clearml integration cleanup * Updated ClearML documentation * Added dark vs light icons ClearML Readme * Clearml Readme styling * Add better gifs * Fixed gif file size * Add better images in tutorial notebook * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Addressed comments in PR #8620 * Fixed circular import * Fixed circular import * Update tutorial.ipynb * Update tutorial.ipynb * Inline comment * Restructured tutorial notebook * Add correct ClearML link to README * Update tutorial.ipynb * Update general.py * Update __init__.py * Update __init__.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update __init__.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update __init__.py * Update README.md * Update __init__.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * spelling * Update tutorial.ipynb * notebook cutt.ly links * Update README.md * Update README.md * cutt.ly links in tutorial * Removed labels as they show up on last subplot only Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-06 02:50:49 +08:00
# clearml
Weights & Biases (W&B) Feature Addition (#1235) * Add wandb metric logging and bounding box debugging * Improve formatting, readability * Remove mutliple path for init, improve formatting * Add wandb params * Remove typecasting in bbox coordinates and reformat * Cleanup * add wandb to requirements.txt * minor updates to test.py * general reorg * reduce --log-imgs to 10 * clean wandb import * reverse wandb import assert * add except AssertionError to try import * move wandb init to all global ranks * replace print() with logger.info() * replace print() with logger.info() * move wandb.init() bug fix * project PosixPath to basename bug fix Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-11-01 03:16:35 +08:00
Add TensorFlow formats to `export.py` (#4479) * Initial commit * Remove unused export_torchscript return * ROOT variable * Add prefix to fcn arg * fix ROOT * check_yaml into run() * interim fixes * imgsz=(320, 320) * Hardcode tf_raw_resize False * Finish opt elimination * Update representative_dataset_gen() * Update export.py with TF methods * SiLU and GraphDef fixes * file_size() directory handling feature * export fixes * add lambda: to representative_dataset * Detect training False default * Fuse false for TF models * Embed agnostic NMS arguments * Remove lambda * TensorFlow.js export success * Add pb to Usage * Add *_tfjs_model/ to ignore files * prepend YOLOv5 to function headers * Remove end --- comments * parameterize tfjs export pb file * update run() data default /ROOT * update --include help * update imports * return ct_model * Consolidate TFLite export * pb prerequisite to tfjs * TF modules CamelCase * Remove exports from tf.py and cleanup * pass agnostic NMS arguments * CI * CI * ignore *_web_model/ * Add tensorflow to CI dependencies * CI tensorflow-cpu * Update requirements.txt * Remove tensorflow check_requirement * CI coreml tfjs * export only onnx torchscript * reorder exports torchscript first
2021-09-12 21:52:24 +08:00
# Plotting ------------------------------------
Add `requests` to requirements.txt (#5112) * Requirements updated 1. Requests added to requirements.txt. That might not be included in all docker base images, adding it to the requirements is safer. 2. Added a minimum version to Pandas. It's a good practice to have versions for all dependencies. * Sort alphabetically * Update requirements.txt Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-10 06:51:01 +08:00
pandas>=1.1.4
Add `roboflow` (#4956)
2021-09-28 10:00:05 +08:00
seaborn>=0.11.0
update requirements.txt
2020-08-15 01:14:48 +08:00
Add TensorFlow formats to `export.py` (#4479) * Initial commit * Remove unused export_torchscript return * ROOT variable * Add prefix to fcn arg * fix ROOT * check_yaml into run() * interim fixes * imgsz=(320, 320) * Hardcode tf_raw_resize False * Finish opt elimination * Update representative_dataset_gen() * Update export.py with TF methods * SiLU and GraphDef fixes * file_size() directory handling feature * export fixes * add lambda: to representative_dataset * Detect training False default * Fuse false for TF models * Embed agnostic NMS arguments * Remove lambda * TensorFlow.js export success * Add pb to Usage * Add *_tfjs_model/ to ignore files * prepend YOLOv5 to function headers * Remove end --- comments * parameterize tfjs export pb file * update run() data default /ROOT * update --include help * update imports * return ct_model * Consolidate TFLite export * pb prerequisite to tfjs * TF modules CamelCase * Remove exports from tf.py and cleanup * pass agnostic NMS arguments * CI * CI * ignore *_web_model/ * Add tensorflow to CI dependencies * CI tensorflow-cpu * Update requirements.txt * Remove tensorflow check_requirement * CI coreml tfjs * export only onnx torchscript * reorder exports torchscript first
2021-09-12 21:52:24 +08:00
# Export --------------------------------------
Update `coremltools>=6.0` (#9532) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-21 23:55:25 +08:00
# coremltools>=6.0 # CoreML export
Add TensorFlow formats to `export.py` (#4479) * Initial commit * Remove unused export_torchscript return * ROOT variable * Add prefix to fcn arg * fix ROOT * check_yaml into run() * interim fixes * imgsz=(320, 320) * Hardcode tf_raw_resize False * Finish opt elimination * Update representative_dataset_gen() * Update export.py with TF methods * SiLU and GraphDef fixes * file_size() directory handling feature * export fixes * add lambda: to representative_dataset * Detect training False default * Fuse false for TF models * Embed agnostic NMS arguments * Remove lambda * TensorFlow.js export success * Add pb to Usage * Add *_tfjs_model/ to ignore files * prepend YOLOv5 to function headers * Remove end --- comments * parameterize tfjs export pb file * update run() data default /ROOT * update --include help * update imports * return ct_model * Consolidate TFLite export * pb prerequisite to tfjs * TF modules CamelCase * Remove exports from tf.py and cleanup * pass agnostic NMS arguments * CI * CI * ignore *_web_model/ * Add tensorflow to CI dependencies * CI tensorflow-cpu * Update requirements.txt * Remove tensorflow check_requirement * CI coreml tfjs * export only onnx torchscript * reorder exports torchscript first
2021-09-12 21:52:24 +08:00
# onnx>=1.9.0 # ONNX export
Upgrade onnxsim to v0.4.1 (#8632) * upgrade onnxsim to v0.4.1 Signed-off-by: daquexian <daquexian566@gmail.com> * Update export.py * Update export.py * Update export.py * Update export.py * Update export.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-07-20 01:01:24 +08:00
# onnx-simplifier>=0.4.1 # ONNX simplifier
Update requirements.txt comment spacing (#8562)
2022-07-15 22:01:29 +08:00
# nvidia-pyindex # TensorRT export
# nvidia-tensorrt # TensorRT export
Update scikit-learn constraint per coremltools 6.0 (#9531) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-21 21:14:54 +08:00
# scikit-learn<=1.1.2 # CoreML quantization
TensorFlow macOS AutoUpdate (#9471) * TensorFlow macOS AutoUpdate * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-18 23:34:34 +08:00
# tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos)
Add TensorFlow formats to `export.py` (#4479) * Initial commit * Remove unused export_torchscript return * ROOT variable * Add prefix to fcn arg * fix ROOT * check_yaml into run() * interim fixes * imgsz=(320, 320) * Hardcode tf_raw_resize False * Finish opt elimination * Update representative_dataset_gen() * Update export.py with TF methods * SiLU and GraphDef fixes * file_size() directory handling feature * export fixes * add lambda: to representative_dataset * Detect training False default * Fuse false for TF models * Embed agnostic NMS arguments * Remove lambda * TensorFlow.js export success * Add pb to Usage * Add *_tfjs_model/ to ignore files * prepend YOLOv5 to function headers * Remove end --- comments * parameterize tfjs export pb file * update run() data default /ROOT * update --include help * update imports * return ct_model * Consolidate TFLite export * pb prerequisite to tfjs * TF modules CamelCase * Remove exports from tf.py and cleanup * pass agnostic NMS arguments * CI * CI * ignore *_web_model/ * Add tensorflow to CI dependencies * CI tensorflow-cpu * Update requirements.txt * Remove tensorflow check_requirement * CI coreml tfjs * export only onnx torchscript * reorder exports torchscript first
2021-09-12 21:52:24 +08:00
# tensorflowjs>=3.9.0 # TF.js export
OpenVINO Export (#6057) * OpenVINO export * Remove timeout * Add 3 files * str * Constrain opset to 12 * Default ONNX opset to 12 * Make dir * Make dir * Cleanup * Cleanup * check_requirements(('openvino-dev',))
2021-12-23 03:29:48 +08:00
# openvino-dev # OpenVINO export
update requirements.txt
2020-08-15 01:14:48 +08:00
Add TensorFlow formats to `export.py` (#4479) * Initial commit * Remove unused export_torchscript return * ROOT variable * Add prefix to fcn arg * fix ROOT * check_yaml into run() * interim fixes * imgsz=(320, 320) * Hardcode tf_raw_resize False * Finish opt elimination * Update representative_dataset_gen() * Update export.py with TF methods * SiLU and GraphDef fixes * file_size() directory handling feature * export fixes * add lambda: to representative_dataset * Detect training False default * Fuse false for TF models * Embed agnostic NMS arguments * Remove lambda * TensorFlow.js export success * Add pb to Usage * Add *_tfjs_model/ to ignore files * prepend YOLOv5 to function headers * Remove end --- comments * parameterize tfjs export pb file * update run() data default /ROOT * update --include help * update imports * return ct_model * Consolidate TFLite export * pb prerequisite to tfjs * TF modules CamelCase * Remove exports from tf.py and cleanup * pass agnostic NMS arguments * CI * CI * ignore *_web_model/ * Add tensorflow to CI dependencies * CI tensorflow-cpu * Update requirements.txt * Remove tensorflow check_requirement * CI coreml tfjs * export only onnx torchscript * reorder exports torchscript first
2021-09-12 21:52:24 +08:00
# Extras --------------------------------------
Add `psutil` and `ipython` to requirements.txt (#7998) * Add `psutil` and `ipython` to requirements.txt Lightweight packages used by YOLOv5 for system utilization (psutil) and interactive notebooks (IPython) * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * sort alphabetically Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-05-27 01:12:19 +08:00
ipython # interactive notebook
psutil # system utilization
Update `tqdm>=4.64.0` and `thop>=0.1.1` (#8596) * Update `tqdm>=4.64.0` and `thop>=0.1.1` * Update requirements.txt
2022-07-16 22:14:16 +08:00
thop>=0.1.1 # FLOPs computation
Add `roboflow` (#4956)
2021-09-28 10:00:05 +08:00
# albumentations>=1.0.3
On-demand `pycocotools` pip install (#3547)
2021-06-09 17:22:21 +08:00
# pycocotools>=2.0 # COCO mAP
Add `roboflow` (#4956)
2021-09-28 10:00:05 +08:00
# roboflow