Skip to content

Latest commit

 

History

History
21 lines (17 loc) · 485 Bytes

README.md

File metadata and controls

21 lines (17 loc) · 485 Bytes

Torch_yolov3

About:

  • A pytorch implementation of yoloV3.
  • Inspired by keras-yolo3

Installation:

  • Download YoloV3 pytorch weight yolo_body.pth from Google Drive
  • Put it under model_data folder
  • Run tests

Requirements

  • pytorch
  • pytest
  • torchvision>=0.4.2
  • numpy
  • Pillow>=6.2.1
  • matplotlib

Test

python -m pytest src/tests/