Example project to load ONNX Yolov3 model and infer some boxes in Scala.
-
Clone this repository https://github.com/novakov-alexey/yolo-keras
-
Go into the repository root folder
-
Create environment via
make
command (Makefile is provided):
make install
-
Download weights from here: https://pjreddie.com/media/files/yolov3.weights and put them into
yolo-keras/models
directory -
Run this script to create Keras model:
python create_model.py
- Run the following command to create ONNX model:
make save-2-onnx
- Copy ONNX model file to the
yolo-scala/models
directory of theyolo-scala
repository