|
|
|
@@ -13,3 +13,31 @@ torch>=1.7.0 |
|
|
|
torchvision>=0.8.1 |
|
|
|
tqdm>=4.64.0 |
|
|
|
protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012 |
|
|
|
|
|
|
|
# Logging ------------------------------------- |
|
|
|
tensorboard>=2.4.1 |
|
|
|
# wandb |
|
|
|
# clearml |
|
|
|
|
|
|
|
# Plotting ------------------------------------ |
|
|
|
pandas>=1.1.4 |
|
|
|
seaborn>=0.11.0 |
|
|
|
|
|
|
|
# Export -------------------------------------- |
|
|
|
# coremltools>=5.2 # CoreML export |
|
|
|
# onnx>=1.9.0 # ONNX export |
|
|
|
# onnx-simplifier>=0.4.1 # ONNX simplifier |
|
|
|
# nvidia-pyindex # TensorRT export |
|
|
|
# nvidia-tensorrt # TensorRT export |
|
|
|
# scikit-learn==0.19.2 # CoreML quantization |
|
|
|
# tensorflow>=2.4.1 # TFLite export (or tensorflow-cpu, tensorflow-aarch64) |
|
|
|
# tensorflowjs>=3.9.0 # TF.js export |
|
|
|
# openvino-dev # OpenVINO export |
|
|
|
|
|
|
|
# Extras -------------------------------------- |
|
|
|
ipython # interactive notebook |
|
|
|
psutil # system utilization |
|
|
|
thop>=0.1.1 # FLOPs computation |
|
|
|
# albumentations>=1.0.3 |
|
|
|
# pycocotools>=2.0 # COCO mAP |
|
|
|
# roboflow |