Alex Stoken
|
de191655e4
|
Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp dict accessor
|
2020-06-24 17:21:54 -05:00 |
|
Alex Stoken
|
2d396bea00
|
Fix bug in --help from percent sign in help string
|
2020-06-24 16:57:12 -05:00 |
|
Alex Stoken
|
611aacf1bf
|
Turn opt into dictionary before sending it to yaml
|
2020-06-24 10:49:08 -05:00 |
|
Alex Stoken
|
bc4ef4861b
|
Default optimizer SGD
|
2020-06-24 10:07:43 -05:00 |
|
Alex Stoken
|
7abf202cad
|
Mode all optimizer settings to 'hyp.yaml', integrate proper momentum with Adam optimizer
|
2020-06-24 10:03:21 -05:00 |
|
Alex Stoken
|
7edbf6570e
|
Fix help message for cfg files
|
2020-06-24 09:45:57 -05:00 |
|
Alex Stoken
|
d64ad0fbf3
|
Remove --resume functionality and related checks/logic.
|
2020-06-24 09:17:27 -05:00 |
|
Alex Stoken
|
e572bb0803
|
Add plot_results save location to log_dir
|
2020-06-21 09:36:28 -05:00 |
|
Alex Stoken
|
c8152c81a6
|
Syntax fixes
|
2020-06-17 16:32:13 -05:00 |
|
Alex Stoken
|
9b7386f603
|
Add save_dir arg to test.test, use arg as location for saving batch jpgs
|
2020-06-17 16:08:46 -05:00 |
|
Alex Stoken
|
945307beba
|
Add save_dir to plot_lr_scheduler and plot_labels
Set save_dir = log_dir in train.py
|
2020-06-17 16:03:18 -05:00 |
|
Alex Stoken
|
3b2b330872
|
Move results.txt from weights/ to log_dir
|
2020-06-17 15:55:45 -05:00 |
|
Alex Stoken
|
ade023cff2
|
Fix hyp file read in and dict update.
Add example of hyp yaml
|
2020-06-17 10:59:20 -05:00 |
|
Alex Stoken
|
5f2eeba233
|
remove old print statements
|
2020-06-16 17:09:39 -05:00 |
|
Alex Stoken
|
333f678b37
|
add update default hyp dict with provided yaml
|
2020-06-16 16:36:20 -05:00 |
|
Alex Stoken
|
a448c3bcd7
|
add logic for resuming and getting hyp for resume run
|
2020-06-16 16:30:12 -05:00 |
|
Alex Stoken
|
25e51bcec7
|
add util function to get most recent last.pt file
added logic in train.py __main__ to handle resuming from a run
|
2020-06-16 15:50:27 -05:00 |
|
Alex Stoken
|
490f1e7b9c
|
add save_dir arg to plot_lr_scheduler, default to current dir.
Uncomment plot_lr_scheduler in train() and pass log_dir as save location
|
2020-06-16 15:13:03 -05:00 |
|
Alex Stoken
|
4418809cf5
|
change weights dir (wdir) to be unique to each run, under log_dir
|
2020-06-16 15:09:51 -05:00 |
|
Alex Stoken
|
d9f446cd81
|
add save yaml of opt and hyp to tensorboard log_dir in train()
|
2020-06-16 15:06:13 -05:00 |
|
Alex Stoken
|
a85e6d0fc0
|
add parser arg for hyp yaml file
|
2020-06-16 14:53:32 -05:00 |
Glenn Jocher
|
8b26e89006
|
AutoAnchor bug fix #72
|
2020-06-16 00:53:34 -07:00 |
|
Glenn Jocher
|
bdd9fee841
|
update fast mode
|
2020-06-16 00:29:54 -07:00 |
|
Glenn Jocher
|
1c0b6236e3
|
update fast mode
|
2020-06-16 00:11:29 -07:00 |
|
Glenn Jocher
|
915b1481fc
|
default check_git_status() to True
|
2020-06-15 16:18:46 -07:00 |
|
Glenn Jocher
|
14523bb030
|
FP16 to FP32 ckpt load
|
2020-06-15 13:18:39 -07:00 |
|
Glenn Jocher
|
c5966abba8
|
glob search bug fix #77
|
2020-06-15 12:08:57 -07:00 |
|
Glenn Jocher
|
31f3310029
|
assert best possible recall > 0.9 before training
|
2020-06-13 15:05:41 -07:00 |
Lauritzen Kasper Primdal
|
c3d4d321d3
|
Ensures weights/ dir exists
Allows train.py to be run outside of yolov5/ directory.
|
2020-06-13 14:44:23 +02:00 |
|
Glenn Jocher
|
099e6f5ebd
|
--img-size stride-multiple verification
|
2020-06-12 22:10:46 -07:00 |
|
Glenn Jocher
|
22d6088205
|
speed-reproducibility fix #17
|
2020-06-05 13:07:09 -07:00 |
|
Glenn Jocher
|
55ca5c74d2
|
multi-scale fix #16
|
2020-06-05 12:57:16 -07:00 |
|
Glenn Jocher
|
7c2832cd49
|
assert equal model and dataset classes
|
2020-06-04 17:21:22 -07:00 |
|
Glenn Jocher
|
11121e39ed
|
updates
|
2020-06-04 16:52:07 -07:00 |
|
Glenn Jocher
|
eb97b2e413
|
NMS fast mode
|
2020-06-03 13:02:59 -07:00 |
|
Glenn Jocher
|
ce36905358
|
updates
|
2020-05-30 00:12:45 -07:00 |
|
Glenn Jocher
|
1e84a23f38
|
initial commit
|
2020-05-29 17:04:54 -07:00 |