Skip to content

novakov-alexey-zz/yolo-scala

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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):

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

  2. Run this script to create Keras model:

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

About

Deep Learning in Scala: Yolo v3 inference

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages