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
3b2b330872
Move results.txt from weights/ to log_dir
2020-06-17 15:55:45 -05:00
ade023cff2
Fix hyp file read in and dict update.
...
Add example of hyp yaml
2020-06-17 10:59:20 -05:00
2368603484
fix refrence bug
...
In torch==1.5, the import of the API has changed. Although it does not interrupt the operation of the program, it seems to me to be an implicit error and may throw an exception in later versions.
2020-06-17 09:56:26 +08:00
8db51c7002
tb_writer bug fix
2020-06-16 16:05:28 -07:00
5f2eeba233
remove old print statements
2020-06-16 17:09:39 -05:00
afe1df385b
dist.destroy_process_group() bug fix
2020-06-16 15:08:14 -07:00
333f678b37
add update default hyp dict with provided yaml
2020-06-16 16:36:20 -05:00
a448c3bcd7
add logic for resuming and getting hyp for resume run
2020-06-16 16:30:12 -05:00
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
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
4418809cf5
change weights dir (wdir) to be unique to each run, under log_dir
2020-06-16 15:09:51 -05:00
d9f446cd81
add save yaml of opt and hyp to tensorboard log_dir in train()
2020-06-16 15:06:13 -05:00
a85e6d0fc0
add parser arg for hyp yaml file
2020-06-16 14:53:32 -05:00
5a50491fa1
check_anchors bug fix
2020-06-16 10:36:35 -07:00
05b8ee5ca4
check_anchors() bug fix #102
2020-06-16 10:34:16 -07:00
ec81c7b5f2
check_anchors() bug fix #90
2020-06-16 10:14:04 -07:00
8b26e89006
AutoAnchor bug fix #72
2020-06-16 00:53:34 -07:00
bdd9fee841
update fast mode
2020-06-16 00:29:54 -07:00
1c0b6236e3
update fast mode
2020-06-16 00:11:29 -07:00
915b1481fc
default check_git_status() to True
2020-06-15 16:18:46 -07:00
14523bb030
FP16 to FP32 ckpt load
2020-06-15 13:18:39 -07:00
c5966abba8
glob search bug fix #77
2020-06-15 12:08:57 -07:00
31f3310029
assert best possible recall > 0.9 before training
2020-06-13 15:05:41 -07:00
c3d4d321d3
Ensures weights/ dir exists
...
Allows train.py to be run outside of yolov5/ directory.
2020-06-13 14:44:23 +02:00
099e6f5ebd
--img-size stride-multiple verification
2020-06-12 22:10:46 -07:00
22d6088205
speed-reproducibility fix #17
2020-06-05 13:07:09 -07:00
55ca5c74d2
multi-scale fix #16
2020-06-05 12:57:16 -07:00
7c2832cd49
assert equal model and dataset classes
2020-06-04 17:21:22 -07:00
11121e39ed
updates
2020-06-04 16:52:07 -07:00
eb97b2e413
NMS fast mode
2020-06-03 13:02:59 -07:00
ce36905358
updates
2020-05-30 00:12:45 -07:00
1e84a23f38
initial commit
2020-05-29 17:04:54 -07:00
Alex Stoken
Lornatang
Glenn Jocher
Lauritzen Kasper Primdal