|
|
|
@ -10,7 +10,7 @@ pyyaml
|
|
|
|
|
torchvision
|
|
|
|
|
scipy
|
|
|
|
|
tqdm
|
|
|
|
|
'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
|
|
|
|
|
git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI
|
|
|
|
|
|
|
|
|
|
# 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
|
|
|
|
|