/PyTorch-YOLOv3

Minimal PyTorch implementation of YOLOv3

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

This is a fork of https://github.com/eriklindernoren/PyTorch-YOLOv3.

Our modifications:

  • Remove tensorflow dependency
  • Add setup.py
  • Change cli interface to click
  • Modifiy detect.py such that is writes the detection results as a pickled pandas dataframe or into a sqlite3 database