2907 Commits

Author SHA1 Message Date
Glenn Jocher
b8557f87e3 add stride to datasets.py 2020-06-24 13:02:27 -07:00
Glenn Jocher
3b062254a6 update utils.create_pretrained() 2020-06-24 12:53:43 -07:00
ChristopherSTAN
d18adf8df9
Add files via upload 2020-06-24 15:53:34 -04:00
Glenn Jocher
0825cb7fd8 create models/hub 2020-06-24 11:58:23 -07:00
Glenn Jocher
b1b3634145 update onnx_export.py #134 2020-06-24 11:00:03 -07:00
Glenn Jocher
8699c319ca update yolo.py 2020-06-24 10:42:01 -07:00
Alex Stoken
d1ca6f231d
Delete equip_hyp.yaml 2020-06-24 11:22:12 -05: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
Alex Stoken
ccf0af18b1 Revert coco128.yaml to initial commit 2020-06-24 09:13:41 -05:00
Glenn Jocher
f9ae460eec update README.md 2020-06-23 20:12:23 -07:00
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
Alex Stoken
e572bb0803 Add plot_results save location to log_dir 2020-06-21 09:36:28 -05: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
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