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
2,768 Commits 10 Branches 10 Tags
Commit Graph

416 Commits

Author SHA1 Message Date
Glenn Jocher
3bdea3f697 strip_optimizer() bug fix #253 2020-07-02 21:24:26 -07:00
Glenn Jocher
e02a189a3a
Merge pull request #245 from yxNONG/patch-2 
Unify the check point of single and multi GPU
 2020-07-02 12:05:46 -07:00
Glenn Jocher
597ed4ce63
Update train.py 2020-07-02 12:00:55 -07:00
Glenn Jocher
13f69777a6 typo fix 2020-07-02 09:26:03 -07:00
yxNONG
1aa2b67933
Update train.py 2020-07-02 13:51:52 +08:00
Glenn Jocher
86784cfdbf --resume bug fix #252 2020-06-30 21:43:53 -07:00
Glenn Jocher
ad4c22cbfe --resume bug fix #187 2020-06-30 16:16:29 -07:00
Glenn Jocher
3b16c865f0 assert --epochs 2020-06-30 14:08:08 -07:00
yxNONG
cdb9bde181
Unify the check point of single and multi GPU 
save the model.hyp etc to checkpoint when use multi GPU training
 2020-06-30 19:06:28 +08:00
Glenn Jocher
b203c9b7ff update train.py incompatible model message fix #222 2020-06-29 12:45:25 -07:00
Glenn Jocher
37e13f8846 update mosaic border 2020-06-27 13:50:15 -07:00
Alex Stoken
e18e6811dc
Merge branch 'master' into advanced_logging 2020-06-27 10:13:03 -05:00
Glenn Jocher
22fb2b0c25 refactor dataloader 2020-06-26 18:56:13 -07:00
Glenn Jocher
256a3e89d2 small dataset bug fix #140 2020-06-25 17:52:56 -07:00
Glenn Jocher
b50fdf16af model.names multi-GPU bug fix #94 2020-06-24 22:22:13 -07:00
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
Glenn Jocher
b8557f87e3 add stride to datasets.py 2020-06-24 13:02:27 -07: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
Glenn Jocher
6c1b87a42e update google_utils import 2020-06-22 23:00:23 -07:00
Glenn Jocher
1f1917ef56 remove fast, add merge 2020-06-21 13:37:11 -07:00
Alex Stoken
e572bb0803 Add plot_results save location to log_dir 2020-06-21 09:36:28 -05:00
Lornatang
899f1d4bde Fix DDP bug in single process multiple device use cases 2020-06-20 13:00:03 +08:00
Glenn Jocher
cdf1eac9f7
Merge pull request #107 from Lornatang/fix-reference-bugs 
fix refrence bug
 2020-06-19 14:51:41 -07:00
Glenn Jocher
cce95e744d backbone as FP16, save default to FP32 2020-06-18 00:13:18 -07:00
Glenn Jocher
d9b64c27c2 save ckpt in FP16 #119 2020-06-17 22:34:13 -07:00
Glenn Jocher
9fdb0fbacf AutoAnchor bug fix # 117 2020-06-17 19:51:15 -07: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
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