#!/usr/bin/env bash
# Copyright 2023-2024 NXP
# SPDX-License-Identifier: MIT

set -e

wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.weights
wget https://raw.githubusercontent.com/amikelive/coco-labels/master/coco-labels-2014_2017.txt


# tensorflow -> tflite
python3.8 -m venv env
source ./env/bin/activate

pip install --upgrade pip
pip install tensorflow==2.10.0
pip install Pillow

wget --no-check-certificate https://images.cocodataset.org/zips/val2017.zip
unzip val2017.zip

# convert model from darknet to tensorflow lite
python3.8 export_model.py --weights_path=./ --output_path=./ --images_path=val2017

# install vela
pip install numpy==1.20
pip install git+https://github.com/nxp-imx/ethos-u-vela.git@lf-6.1.22-2.0.0

vela --output-dir model_imx93 yolov4-tiny_416_quant.tflite

# cleanup 
deactivate
rm -rf val2017 env
rm val2017.zip
rm yolov4-tiny.weights