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

Now include json in model zip along with onnx file and .pt file

This commit is contained in:
mmcwilliams 2023-12-15 12:15:33 -05:00
parent 308f1174ff
commit 87c6540a80
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/train.sh
View File

@ -49,7 +49,7 @@ if [ ! -f "${UNZIPPED}/data.yaml" ]; then
fi fi
FULLDEST=$(realpath "${UNZIPPED}") FULLDEST=$(realpath "${UNZIPPED}")
python scripts/data.py "${FULLDEST}/data.yaml" python scripts/data.py "${FULLDEST}/data.yaml" "${NAME}"
cd yolov5 cd yolov5
source env/bin/activate source env/bin/activate
@ -84,11 +84,13 @@ fi
MODELPATH=$(realpath "../models/${ID}/weights/best.onnx") MODELPATH=$(realpath "../models/${ID}/weights/best.onnx")
PTPATH=$(realpath "../models/${ID}/weights/best.pt") PTPATH=$(realpath "../models/${ID}/weights/best.pt")
JSONPATH=$(realpath "${FULLDEST}/${NAME}.json")
ZIPPATH="${ID}.zip" ZIPPATH="${ID}.zip"
mkdir zip mkdir zip
cp "${MODELPATH}" "zip/${NAME}.onnx" cp "${MODELPATH}" "zip/${NAME}.onnx"
cp "${PTPATH}" "zip/${NAME}.pt" cp "${PTPATH}" "zip/${NAME}.pt"
cp "${JSONPATH}" "zip/${NAME}.json"
cd zip cd zip
zip "../${ZIPPATH}" * zip "../${ZIPPATH}" *
cd .. cd ..