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

43 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
Update protobuf requirement from <=3.20.1 to <4.21.3 (#8346) Updates the requirements on [protobuf](https://github.com/protocolbuffers/protobuf) to permit the latest version. - [Release notes](https://github.com/protocolbuffers/protobuf/releases) - [Changelog](https://github.com/protocolbuffers/protobuf/blob/main/generate_changelog.py) - [Commits](https://github.com/protocolbuffers/protobuf/commits) --- updated-dependencies: - dependency-name: protobuf dependency-type: direct:production ... Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-06-28 18:04:58 +08:00
protobuf<4.21.3 # 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
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 --------------------------------------
# coremltools>=4.1 # CoreML export
# onnx>=1.9.0 # ONNX export
# onnx-simplifier>=0.3.6 # ONNX simplifier
Update requirements.txt comment spacing (#8562)
2022-07-15 22:01:29 +08:00
# nvidia-pyindex # TensorRT export
# nvidia-tensorrt # TensorRT 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
# scikit-learn==0.19.2 # CoreML quantization
# tensorflow>=2.4.1 # TFLite export
# 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