From 4d49957f5a0a04db1cda1d9306aeb47c48dda0d1 Mon Sep 17 00:00:00 2001 From: e96031413 <30921855+e96031413@users.noreply.github.com> Date: Fri, 11 Sep 2020 03:00:53 +0800 Subject: [PATCH] Update requirements.txt (#1481) * Update requirements.txt I found that if we would like to calculate FLOPS in this project, we must install thop. but there's no thop package inside the requirements.txt https://github.com/ultralytics/yolov3/blob/master/utils/torch_utils.py#L108 * Update requirements.txt Co-authored-by: Glenn Jocher --- requirements.txt | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/requirements.txt b/requirements.txt index 895cf6ca..50efbd10 100755 --- a/requirements.txt +++ b/requirements.txt @@ -1,10 +1,11 @@ # pip install -r requirements.txt + +# base ---------------------------------------- Cython matplotlib>=3.2.2 numpy>=1.18.5 opencv-python>=4.1.2 pillow -# pycocotools>=2.0 PyYAML>=5.3 scipy>=1.4.1 tensorboard>=2.2 @@ -12,10 +13,15 @@ torch>=1.6.0 torchvision>=0.7.0 tqdm>=4.41.0 -# Conda commands (in place of pip) --------------------------------------------- -# conda update -yn base -c defaults conda -# conda install -yc anaconda numpy opencv matplotlib tqdm pillow ipython -# conda install -yc conda-forge scikit-image pycocotools tensorboard -# conda install -yc spyder-ide spyder-line-profiler -# conda install -yc pytorch pytorch torchvision -# conda install -yc conda-forge protobuf numpy && pip install onnx==1.6.0 # https://github.com/onnx/onnx#linux-and-macos +# coco ---------------------------------------- +# pycocotools>=2.0 + +# export -------------------------------------- +# packaging # for coremltools +# coremltools==4.0b3 +# onnx>=1.7.0 +# scikit-learn==0.19.2 # for coreml quantization + +# extras -------------------------------------- +# thop # FLOPS computation +# seaborn # plotting