You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1.0 KiB

测试环境:
机型联想拯救者R7000 2020
CPU: AMD R-4800H 
GPU: RTX 2060(6G) GTX 1650(6G)
依赖:

# pip install -r requirements.txt

# Base ----------------------------------------

matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.2
Pillow>=7.1.2
PyYAML>=5.3.1
requests>=2.23.0
scipy>=1.4.1
torch>=1.7.0
torchvision>=0.8.1
tqdm>=4.41.0
pyqt5

# Logging -------------------------------------

tensorboard>=2.4.1

# wandb

# Plotting ------------------------------------

pandas>=1.1.4
seaborn>=0.11.0

# Export --------------------------------------

# coremltools>=4.1  # CoreML export

# onnx>=1.9.0  # ONNX export

# onnx-simplifier>=0.3.6  # ONNX simplifier

# scikit-learn==0.19.2  # CoreML quantization

# tensorflow>=2.4.1  # TFLite export

# tensorflowjs>=3.9.0  # TF.js export

# openvino-dev  # OpenVINO export

# Extras --------------------------------------

# albumentations>=1.0.3

# Cython  # for pycocotools https://github.com/cocodataset/cocoapi/issues/172

# pycocotools>=2.0  # COCO mAP

# roboflow

thop  # FLOPs computation