Commit Graph

156 Commits (cdf1eac9f7d7d714eb79f42dd7bc664e23c2eb25)
 

Author SHA1 Message Date
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
Glenn Jocher 57a0ae3350 AutoAnchor implementation 2020-06-16 13:12:22 -07:00
Glenn Jocher 95c46f7245
Merge pull request #79 from Lornatang/delete-redundant-thrid-party-library
Delete redundant thrid party library
2020-06-16 11:52:36 -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 7c6fd4d670 --verbose bug fix #84 2020-06-16 10:22:22 -07:00
Glenn Jocher ec81c7b5f2 check_anchors() bug fix #90 2020-06-16 10:14:04 -07:00
Glenn Jocher e1e3399286 ONNX export bug fix #93 2020-06-16 09:59:42 -07:00
Glenn Jocher 8b26e89006 AutoAnchor bug fix #72 2020-06-16 00:53:34 -07:00
Glenn Jocher 8fa3724072 kmean_anchors() update 2020-06-16 00:40:21 -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
Lornatang 8b8b792802 There is no need to download extra packages, official bring it with you
I submitted it once in your yolov3 project, you seem to accept it? I'm not sure. I'll submit PR again.
2020-06-16 08:28:52 +08:00
Lornatang 0c4b4b8817 Merge remote-tracking branch 'upstream/master' 2020-06-16 08:16:10 +08:00
Lornatang 09c1b96136 Merge branch 'fix-reference-bugs' 2020-06-16 08:15:10 +08:00
Glenn Jocher db2c3acd3a updated testing settings, rebalanced towards FP16 latency 2020-06-15 16:55:57 -07:00
Glenn Jocher 915b1481fc default check_git_status() to True 2020-06-15 16:18:46 -07:00
Glenn Jocher 0b514daced FP16 test loss bug fix 2020-06-15 15:26:29 -07:00
Glenn Jocher a1748a8d6e test during training default to FP16 2020-06-15 15:20:27 -07:00
Glenn Jocher bd3e389130 update google_utils.py 2020-06-15 14:39:04 -07:00
Glenn Jocher 76ca367a01 FP16 to FP32 ckpt load 2020-06-15 13:32:45 -07:00
Glenn Jocher 6fb5ff014c FP16 to FP32 ckpt load 2020-06-15 13:21:34 -07:00