mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
1,792 Commits 10 Branches 10 Tags
Commit Graph

335 Commits

Author SHA1 Message Date
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
Lornatang
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
Glenn Jocher
8db51c7002 tb_writer bug fix 2020-06-16 16:05:28 -07:00
Alex Stoken
5f2eeba233 remove old print statements 2020-06-16 17:09:39 -05:00
Glenn Jocher
afe1df385b dist.destroy_process_group() bug fix 2020-06-16 15:08:14 -07: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
5a50491fa1 check_anchors bug fix 2020-06-16 10:36:35 -07:00
Glenn Jocher
05b8ee5ca4 check_anchors() bug fix #102 2020-06-16 10:34:16 -07:00
Glenn Jocher
ec81c7b5f2 check_anchors() bug fix #90 2020-06-16 10:14:04 -07: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