2020-05-30 08:04:54 +08:00
|
|
|
# pip install -U -r requirements.txt
|
2020-06-14 16:18:31 +08:00
|
|
|
Cython
|
2020-07-15 06:34:54 +08:00
|
|
|
numpy==1.17.3
|
2020-05-30 08:04:54 +08:00
|
|
|
opencv-python
|
2020-07-09 01:21:30 +08:00
|
|
|
torch>=1.5.1
|
2020-05-30 08:04:54 +08:00
|
|
|
matplotlib
|
|
|
|
|
pillow
|
|
|
|
|
tensorboard
|
2020-06-15 09:32:29 +08:00
|
|
|
PyYAML>=5.3
|
2020-06-14 16:18:31 +08:00
|
|
|
torchvision
|
|
|
|
|
scipy
|
2020-06-15 01:23:33 +08:00
|
|
|
tqdm
|
2020-06-15 01:54:00 +08:00
|
|
|
git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI
|
2020-05-30 08:04:54 +08:00
|
|
|
|
|
|
|
|
# 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
|
2020-06-17 16:45:27 +08:00
|
|
|
# conda install -yc conda-forge protobuf numpy && pip install onnx==1.6.0 # https://github.com/onnx/onnx#linux-and-macos
|
