Add yolov5 as submodule and place to put trained files

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "yolov5"]
+	path = yolov5
+	url = https://github.com/ultralytics/yolov5.git

models/.gitignore

@@ -0,0 +1,3 @@
+*
+
+!.gitignore

scripts/install_yolov5.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+
+cd yolov5
+python -m venv env
+source env/bin/activate
+python -m pip install -r requirements.txt

scripts/train.sh

@@ -46,5 +46,19 @@ if [ ! -f "${UNZIPPED}/data.yaml" ]; then
 	exit 1
 fi
 
-FULLDEST=$(realpath "${DEST}")
+FULLDEST=$(realpath "${UNZIPPED}")
 python scripts/data.py "${FULLDEST}/data.yaml"
+
+cd yolov5
+source env/bin/activate
+
+WEIGHTS="yolov5s.pt"
+
+if [[ "${MODEL}" == "yolov5m_onnx" ]]; then
+	WEIGHTS="yolov5m.pt"
+else if [[ "${MODEL}" == "yolov5l_onnx" ]]; then
+	WEIGHTS="yolov5l.pt"
+fi
+
+python train.py --img 640 --batch 16 --epochs 300 --data "${FULLDEST}/data.yaml" --weights "${WEIGHTS}" --cache --project ../models/ --name "${ID}"
+

yolov5

@@ -0,0 +1 @@
+Subproject commit 0004c745daeebb0eb6455b90c3e8abb308e61320