项目作者: novakov-alexey

项目描述 :
Deep Learning in Scala: Yolo v3 inference
高级语言: Scala
项目地址: git://github.com/novakov-alexey/yolo-scala.git
创建时间: 2021-05-31T07:04:46Z
项目社区:https://github.com/novakov-alexey/yolo-scala

开源协议:

下载


Yolo v3 Inference in Scala

Example project to load ONNX Yolov3 model and infer some boxes in Scala.

city-out-image

Get ONNX model for Yolov3

  1. Clone this repository https://github.com/novakov-alexey/yolo-keras

  2. Go into the repository root folder

  3. Create environment via make command (Makefile is provided):

    1. make install
  4. Download weights from here: https://pjreddie.com/media/files/yolov3.weights and put them into yolo-keras/models directory

  5. Run this script to create Keras model:

    1. python create_model.py
  6. Run the following command to create ONNX model:
    1. make save-2-onnx
  7. Copy ONNX model file to the yolo-scala/models directory of the yolo-scala repository