requirements.txt 797 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. # pip install -r requirements.txt
  2. # FastAPI --------------------------------------
  3. fastapi
  4. uvicorn
  5. jinja2
  6. aiofiles
  7. python-multipart
  8. requests
  9. # Other stuff from https://github.com/ultralytics/yolov5/blob/master/requirements.txt
  10. # base ----------------------------------------
  11. Cython
  12. matplotlib>=3.2.2
  13. numpy>=1.18.5
  14. opencv-python>=4.1.2
  15. Pillow
  16. PyYAML>=5.3.1
  17. scipy>=1.4.1
  18. tensorboard>=2.2
  19. torch>=1.7.0
  20. tqdm>=4.41.0
  21. # logging -------------------------------------
  22. # wandb
  23. # plotting ------------------------------------
  24. seaborn>=0.11.0
  25. pandas
  26. # export --------------------------------------
  27. # coremltools>=4.1
  28. # onnx>=1.8.1
  29. # scikit-learn==0.19.2 # for coreml quantization
  30. # extras --------------------------------------
  31. thop # FLOPS computation
  32. pycocotools>=2.0 # COCO mAP
  33. torchvision>=0.8.1