mattmcw
/
yolo_web
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Ready to run train jobs. TODO: export and upload model

This commit is contained in:
Matt McWilliams 2023-06-29 09:28:10 -04:00
parent 559e0777cf
commit 849ae58f9b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/train.sh
View File

@ -37,7 +37,7 @@ if [ ! -f "${DEST}" ]; then
wget -q -O "${DEST}" "${DOWNLOAD}" wget -q -O "${DEST}" "${DOWNLOAD}"
fi fi
unzip "${DEST}" -d "${UNZIPPED}" unzip -o "${DEST}" -d "${UNZIPPED}"
if [ ! -f "${UNZIPPED}/data.yaml" ]; then if [ ! -f "${UNZIPPED}/data.yaml" ]; then
fail "${ID}" "Invalid dataset" fail "${ID}" "Invalid dataset"
@ -56,7 +56,7 @@ WEIGHTS="yolov5s.pt"
if [[ "${MODEL}" == "yolov5m_onnx" ]]; then if [[ "${MODEL}" == "yolov5m_onnx" ]]; then
WEIGHTS="yolov5m.pt" WEIGHTS="yolov5m.pt"
else if [[ "${MODEL}" == "yolov5l_onnx" ]]; then elif [[ "${MODEL}" == "yolov5l_onnx" ]]; then
WEIGHTS="yolov5l.pt" WEIGHTS="yolov5l.pt"
fi fi