forked from nightsnack/YOLObile
-
Notifications
You must be signed in to change notification settings - Fork 0
/
requirements.txt
executable file
·26 lines (22 loc) · 1001 Bytes
/
requirements.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
# pip install -U -r requirements.txt
# pycocotools requires numpy 1.17 https://github.com/cocodataset/cocoapi/issues/356
numpy == 1.17.5
opencv-python >= 4.1
torch == 1.4.0
torchvision==0.5.0
matplotlib
pycocotools
tqdm
pillow
tensorboard >= 1.14
PyYAML
thop
# Nvidia Apex (optional) for mixed precision training --------------------------
# git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex
# 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 # https://github.com/onnx/onnx#linux-and-macos