parent
5d53db54f3
commit
029c3c37e7
@ -0,0 +1,30 @@
|
|||||||
|
# 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
|
Loading…
Reference in new issue