31 Commits

Author SHA1 Message Date
Glenn Jocher
b50fdf16af model.names multi-GPU bug fix #94 2020-06-24 22:22:13 -07:00
Glenn Jocher
b8557f87e3 add stride to datasets.py 2020-06-24 13:02:27 -07: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
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
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
Glenn Jocher
afe1df385b dist.destroy_process_group() bug fix 2020-06-16 15:08:14 -07: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