yolov5/requirements.txt

# pip install -r requirements.txt

# base ----------------------------------------
Cython
matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.2
Pillow
PyYAML>=5.3.1
scipy>=1.4.1
tensorboard>=2.2
torch>=1.7.0
torchvision>=0.8.1
tqdm>=4.41.0

# logging -------------------------------------
# wandb

# plotting ------------------------------------
seaborn>=0.11.0
pandas

# export --------------------------------------
# coremltools>=4.1
# onnx>=1.8.1
# scikit-learn==0.19.2  # for coreml quantization

# extras --------------------------------------
thop  # FLOPS computation
pycocotools>=2.0  # COCO mAP
update requirements.txt 2020-08-09 03:57:05 +08:00			`# pip install -r requirements.txt`
update requirements.txt 2020-08-15 01:14:48 +08:00
			`# base ----------------------------------------`
Fix pip requirements issues 2020-06-14 16:18:31 +08:00			`Cython`
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`
update requirements.txt 2020-08-02 05:00:14 +08:00			`opencv-python>=4.1.2`
PyTorch Hub and autoShape update (#1415) 2020-11-17 06:14:07 +08:00			`Pillow`
Docker pyYAML>=5.3.1 fix (#2031) 2021-01-25 03:54:15 +08:00			`PyYAML>=5.3.1`
update requirements.txt 2020-08-02 05:00:38 +08:00			`scipy>=1.4.1`
update requirements.txt 2020-08-02 05:00:14 +08:00			`tensorboard>=2.2`
Simplify autoshape() post-process (#1653) * Simplify autoshape() post-process * cleanup * cleanup 2020-12-09 23:44:06 +08:00			`torch>=1.7.0`
			`torchvision>=0.8.1`
update requirements.txt 2020-08-02 05:00:14 +08:00			`tqdm>=4.41.0`
initial commit 2020-05-30 08:04:54 +08:00
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			`# logging -------------------------------------`
			`# wandb`

Confusion matrix (#1474) * initial commit * add plotting * matrix to cpu * bug fix * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * seaborn pandas to requirements.txt * seaborn pandas to requirements.txt * update wandb plotting * remove pandas * if plots * if plots * if plots * if plots * if plots * initial commit * add plotting * matrix to cpu * bug fix * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * seaborn pandas to requirements.txt * seaborn pandas to requirements.txt * update wandb plotting * remove pandas * if plots * if plots * if plots * if plots * if plots * Cat apriori to autolabels * cleanup 2020-11-23 22:27:14 +08:00			`# plotting ------------------------------------`
Fix: Seaborn error due to histplot (#1740) 2020-12-20 01:30:43 +08:00			`seaborn>=0.11.0`
Confusion matrix (#1474) * initial commit * add plotting * matrix to cpu * bug fix * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * seaborn pandas to requirements.txt * seaborn pandas to requirements.txt * update wandb plotting * remove pandas * if plots * if plots * if plots * if plots * if plots * initial commit * add plotting * matrix to cpu * bug fix * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * update plot * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * cleanup * seaborn pandas to requirements.txt * seaborn pandas to requirements.txt * update wandb plotting * remove pandas * if plots * if plots * if plots * if plots * if plots * Cat apriori to autolabels * cleanup 2020-11-23 22:27:14 +08:00			`pandas`
update requirements.txt 2020-08-15 01:14:48 +08:00
			`# export --------------------------------------`
Update requirements.txt (#2173) 2021-02-10 14:03:29 +08:00			`# coremltools>=4.1`
			`# onnx>=1.8.1`
Update requirements.txt for coremltools kmeans quantization 2020-08-23 11:50:53 +08:00			`# scikit-learn==0.19.2 # for coreml quantization`
update requirements.txt 2020-08-15 01:14:48 +08:00
			`# extras --------------------------------------`
Simplify autoshape() post-process (#1653) * Simplify autoshape() post-process * cleanup * cleanup 2020-12-09 23:44:06 +08:00			`thop # FLOPS computation`
			`pycocotools>=2.0 # COCO mAP`