forked from ultralytics/yolov5
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_script_npy.py
73 lines (58 loc) · 2.14 KB
/
run_script_npy.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import os
imgsize = 1024
epochs = 1
batch_size = 1
data_ = "../npy_dataset/data.yaml"
hpy_aug = "../npy_dataset/hyp.manual_augment.yaml"
hpy_no_aug = "../npy_dataset/hyp.no_augment.yaml"
'''
No Augmentations
'''
project_name = "./runs/train/npy_version/no_augment"
os.system("python train.py --img {} \
--epochs {} --batch-size {} \
--data {} \
--input-ch 7 --target-ch 1 \
--weights '' --hyp {} --cfg yolov5n.yaml \
--device 0 \
--exist-ok --patience 0 \
--name scratch_5n --project {}".format(imgsize, epochs, batch_size, data_, hpy_no_aug, project_name))
# os.system("python train.py --img {} \
# --epochs {} --batch-size {} \
# --data {} \
# --input-ch 7 --target-ch 3 \
# --weights '' --hyp {} --cfg yolov5s.yaml \
# --device 0 --exist-ok --patience 0 \
# --name scratch_5s --project {}".format(imgsize, epochs, batch_size, data_, hpy_no_aug, project_name))
# os.system("python train.py --img {} \
# --epochs {} --batch-size {} \
# --data {} \
# --input-ch 7 --target-ch 3 \
# --weights '' --hyp {} --cfg yolov5m.yaml \
# --device 0 --exist-ok --patience 0 \
# --name scratch_5m --project {}".format(imgsize, epochs, batch_size, data_, hpy_no_aug, project_name))
# '''
# With Augmentations
# '''
# project_name = "./runs/train/npy_version_YoDa3ch/augment"
# os.system("python train.py --img {} \
# --epochs {} --batch-size {} \
# --data {} \
# --input-ch 7 --target-ch 3 \
# --weights '' --hyp {} --cfg yolov5n.yaml \
# --device 0 --exist-ok --patience 0 \
# --name scratch_5n --project {}".format(imgsize, epochs, batch_size, data_, hpy_aug, project_name))
# os.system("python train.py --img {} \
# --epochs {} --batch-size {} \
# --data {} \
# --input-ch 7 --target-ch 3 \
# --weights '' --hyp {} --cfg yolov5s.yaml \
# --device 0 --exist-ok --patience 0 \
# --name scratch_5s --project {} --resume".format(imgsize, epochs, batch_size, data_, hpy_aug, project_name))
# os.system("python train.py --img {} \
# --epochs {} --batch-size {} \
# --data {} \
# --input-ch 7 --target-ch 3 \
# --weights '' --hyp {} --cfg yolov5m.yaml \
# --device 0 --exist-ok --patience 0 \
# --name scratch_5m --project {}".format(imgsize, epochs, batch_size, data_, hpy_aug, project_name))