1747 Commits

Author SHA1 Message Date
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
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
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
Alex Stoken
ade023cff2 Fix hyp file read in and dict update.
Add example of hyp yaml
2020-06-17 10:59:20 -05:00
Alex Stoken
3263a204ea Fix get_latest_run to search 'search_dir' recursivly 2020-06-17 10:34:37 -05:00
Alex Stoken
d34291733b Fix get_latest_run() to search inside 'weights' subfolders 2020-06-17 10:26:55 -05: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