177 Commits

Author SHA1 Message Date
Glenn Jocher
ec72eea62b Merge remote-tracking branch 'origin/master' 2020-06-23 10:53:48 -07:00
Glenn Jocher
53294f28b4 update yolo.py 2020-06-23 10:53:42 -07:00
Glenn Jocher
6cea1bff60
Update README.md 2020-06-23 10:11:34 -07:00
Glenn Jocher
a63e1c9b3c study update 2020-06-22 23:23:50 -07:00
Glenn Jocher
2a550333c0 update README.md 2020-06-22 23:20:37 -07:00
Glenn Jocher
db898131f9 study update 2020-06-22 23:07:22 -07:00
Glenn Jocher
6c1b87a42e update google_utils import 2020-06-22 23:00:23 -07:00
Glenn Jocher
7306dadf78
Update README.md 2020-06-22 15:16:10 -07:00
Glenn Jocher
3804578b78
Update README.md 2020-06-22 14:33:47 -07:00
Glenn Jocher
69ff486693
Update README.md 2020-06-22 14:32:05 -07:00
Glenn Jocher
364fcfd7db PANet update 2020-06-22 14:27:17 -07:00
Glenn Jocher
ef58dac963 update Dockerfile 2020-06-22 10:24:56 -07:00
Glenn Jocher
bbd12c7cfe detect.py fast bug fix 2020-06-22 10:14:58 -07:00
Glenn Jocher
fc171e2678 check_anchor_order() update 2020-06-21 14:08:59 -07:00
Glenn Jocher
1f1917ef56 remove fast, add merge 2020-06-21 13:37:11 -07:00
Glenn Jocher
24dd150fbd update torch>=1.4 2020-06-20 11:55:39 -07:00
Glenn Jocher
ba340ed72d
Merge pull request #143 from Lornatang/Fix-multi-GPU-training-errors
Fix DDP bug in single process multiple device use cases
2020-06-20 11:10:40 -07:00
Glenn Jocher
024a42f4fd datasets.py vid_formats updated 2020-06-20 09:53:37 -07:00
Glenn Jocher
5c2cd711df vid_formats bug fix #146 2020-06-20 09:26:07 -07:00
Lornatang
899f1d4bde Fix DDP bug in single process multiple device use cases 2020-06-20 13:00:03 +08:00
Glenn Jocher
0a09882e4f Update issue templates 2020-06-19 16:46:21 -07: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
458c4d0e0b
Update README.md 2020-06-19 12:14:06 -07:00
Glenn Jocher
34ec87ecf9
Update README.md 2020-06-19 12:13:15 -07:00
Glenn Jocher
d4c6674c98 speeds updated from FP32 to FP16 2020-06-19 12:05:10 -07:00
Glenn Jocher
e249792c2f update onnx_export.py usage 2020-06-19 10:15:08 -07:00
Glenn Jocher
80b82e8bde update yolo.py 2020-06-18 19:33:02 -07:00
Glenn Jocher
245bd1d242 update README.md 2020-06-18 12:46:30 -07:00
Glenn Jocher
a3ee0a8954 check_img_size() update 2020-06-18 11:46:04 -07:00
Glenn Jocher
10145833c3 strip_optimizer() default to save FP16 model 2020-06-18 00:16:47 -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
Glenn Jocher
8b6f5826bc update README.md 2020-06-17 12:58:01 -07:00
Glenn Jocher
2bfd242fa1 Merge remote-tracking branch 'origin/master' 2020-06-17 12:57:03 -07:00
Glenn Jocher
b7ac446d62 update README.md 2020-06-17 12:56:47 -07:00
Glenn Jocher
91de3381e1
Merge pull request #112 from imprashr/patch-1
onnx should be 1.6.0
2020-06-17 08:23:35 -07:00
Prashant Rai
0071b69994
onnx should be 1.6.0
for smoothly export model to onnx, onnx should be 1.6.0
2020-06-17 14:15:27 +05:30
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
a557b7d924 check_img_size() update 2020-06-16 17:48:01 -07:00
Glenn Jocher
034609414e AutoAnchor fitness to screen 2020-06-16 17:08:45 -07:00
Glenn Jocher
ef75f7313d AutoAnchor n, imgsz bug fix 2020-06-16 16:48:01 -07:00
Glenn Jocher
2daa412575 train with multi-gpu half test bug fix #99 2020-06-16 16:28:25 -07:00
Glenn Jocher
d1e5716610 train with multi-gpu half test bug fix #99 2020-06-16 16:22:58 -07:00
Glenn Jocher
1dfc28527f AutoAnchor BPR to 4 significant figures 2020-06-16 16:19:58 -07:00
Glenn Jocher
8db51c7002 tb_writer bug fix 2020-06-16 16:05:28 -07:00
Glenn Jocher
4052603e1f AutoAnchor update - improvement check 2020-06-16 15:54:20 -07:00
Glenn Jocher
afe1df385b dist.destroy_process_group() bug fix 2020-06-16 15:08:14 -07:00
Glenn Jocher
72d06147b2 Multi-GPU disabled in test.py 2020-06-16 14:42:30 -07:00
Glenn Jocher
bafbc65ee3 AutoAnchor bug fix 2020-06-16 14:26:02 -07:00