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

Run different kinds of models

This commit is contained in:
Matt McWilliams 2023-06-28 23:29:45 -04:00
parent a6cb4ec856
commit a7e9c5cbf3
3 changed files with 69 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
scripts/data.py Normal file
View File

@ -0,0 +1,64 @@
import sys
import tempfile
import os.path
import os
import shutil
print(sys.argv[-1])
if sys.argv[-1] == 'data.py' or 'data.yaml' not in sys.argv[-1]:
print('Please provide a path to a data.yaml file')
exit(1)
filePath = sys.argv[-1]
data = []
basePath = os.path.dirname(filePath)
trainPath = os.path.join(basePath, 'train', 'images')
validPath = os.path.join(basePath, 'valid', 'images')
testPath = os.path.join(basePath, 'test', 'images')
if not os.path.exists(basePath):
print(f'Dataset directory {basePath} doesn\'t exist')
exit(2)
if not os.path.exists(trainPath):
print(f'Dataset directory {trainPath} doesn\'t exist')
exit(2)
if not os.path.exists(validPath):
print(f'Dataset directory {validPath} doesn\'t exist')
exit(2)
if not os.path.exists(testPath):
print(f'Dataset directory {testPath} doesn\'t exist')
exit(2)
def isInList (match, l) :
for i in l:
if match in i:
return True
return False
with open(filePath, 'r') as file:
lines = file.readlines()
for line in lines:
if 'train: ' in line:
data.append(f'train: {trainPath}{os.linesep}')
elif 'val: ' in line:
data.append(f'val: {validPath}{os.linesep}')
elif 'test: ' in line:
data.append(f'test: {testPath}{os.linesep}')
else:
data.append(line)
if not isInList('train: ', data):
data.append(f'train: {trainPath}{os.linesep}')
elif not isInList('val: ', data):
data.append(f'val: {validPath}{os.linesep}')
elif not isInList('test: ', data):
data.append(f'test: {testPath}{os.linesep}')
file.close()
with tempfile.NamedTemporaryFile() as tmp:
with open(tmp.name, 'w') as t:
t.writelines(data)
shutil.copy(tmp.name, filePath)

2
scripts/train.sh
View File

@ -47,4 +47,4 @@ if [ ! -f "${UNZIPPED}/data.yaml" ]; then
fi
FULLDEST=$(realpath "${DEST}")
python scripts/data.py "${FULLDEST}/data.yaml"

4
views/index.html
View File

@ -5,7 +5,9 @@
<body>
<form method="POST" enctype="multipart/form-data">
<select name="model" id="model" required>
<option value="yolov5_onnx">YOLOv5 -> ONNX</option>
<option value="yolov5s_onnx">YOLOv5 (small) -> ONNX</option>
<option value="yolov5m_onnx">YOLOv5 (medium) -> ONNX</option>
<option value="yolov5l_onnx">YOLOv5 (large) -> ONNX</option>
</select>
<br />
<input type="text" name="name" id="name" placeholder="Model name" required />