azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 09:06:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/main'

Browse Source 
# Conflicts:
#	exports/export.py
This commit is contained in:
zxsanny
2025-03-26 17:20:42 +02:00
parent 3e02be7068 7d99e377f1
commit eaef1a9b66
7 changed files with 239 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+10 -9
View File
@@ -216,9 +216,9 @@ def validate(model_path):
pass
def upload_model(model_path: str):
model = YOLO(model_path)
model.export(format="onnx", imgsz=1280, nms=True, batch=4)
def upload_model(model_path: str, size_small_in_kb: int=3):
# model = YOLO(model_path)
# model.export(format="onnx", imgsz=1280, nms=True, batch=4)
onnx_model = path.dirname(model_path) + Path(model_path).stem + '.onnx'
with open(onnx_model, 'rb') as f_in:
@@ -227,7 +227,7 @@ def upload_model(model_path: str):
key = Security.get_model_encryption_key()
onnx_encrypted = Security.encrypt_to(onnx_bytes, key)
part1_size = min(10 * 1024, int(0.9 * len(onnx_encrypted)))
part1_size = min(size_small_in_kb * 1024, int(0.9 * len(onnx_encrypted)))
onnx_part_small = onnx_encrypted[:part1_size] # slice bytes for part1
onnx_part_big = onnx_encrypted[part1_size:]
@@ -243,8 +243,9 @@ def upload_model(model_path: str):
api.upload_file('azaion.onnx.small', onnx_part_small)
if __name__ == '__main__':
model_path = train_dataset(from_scratch=True)
validate(path.join('runs', 'detect', 'train7', 'weights', 'best.pt'))
form_data_sample(500)
convert2rknn()
upload_model('azaion.pt')
# model_path = train_dataset(from_scratch=True)
# validate(path.join('runs', 'detect', 'train7', 'weights', 'best.pt'))
# form_data_sample(500)
# convert2rknn()
upload_model('azaion-2024-10-26.onnx')