This website requires JavaScript.
8f7e11dcef
Update README.md
Glenn Jocher
2020-06-25 11:53:13 -07:00
b2d4307848
Update README.md
Glenn Jocher
2020-06-25 10:59:42 -07:00
b50fdf16af
model.names multi-GPU bug fix #94
Glenn Jocher
2020-06-24 22:22:13 -07:00
9a9c4f1259
update yolo.py default --cfg
Glenn Jocher
2020-06-24 18:00:43 -07:00
de191655e4
Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp dict accessor
Alex Stoken
2020-06-24 17:21:54 -05:00
2d396bea00
Fix bug in --help from percent sign in help string
Alex Stoken
2020-06-24 16:57:12 -05:00
227aa735dc
updates
Glenn Jocher
2020-06-24 13:20:14 -07:00
e071b8dd3a
update models/common.py for Conv() flexible padding
Glenn Jocher
2020-06-24 13:12:23 -07:00
b8557f87e3
add stride to datasets.py
Glenn Jocher
2020-06-24 13:02:27 -07:00
3b062254a6
update utils.create_pretrained()
Glenn Jocher
2020-06-24 12:53:43 -07:00
d18adf8df9
Add files via upload
ChristopherSTAN
2020-06-24 15:53:34 -04:00
0825cb7fd8
create models/hub
Glenn Jocher
2020-06-24 11:58:23 -07:00
b1b3634145
update onnx_export.py #134
Glenn Jocher
2020-06-24 11:00:03 -07:00
8699c319ca
update yolo.py
Glenn Jocher
2020-06-24 10:42:01 -07:00
d1ca6f231d
Delete equip_hyp.yaml
Alex Stoken
2020-06-24 11:22:12 -05:00
611aacf1bf
Turn opt into dictionary before sending it to yaml
Alex Stoken
2020-06-24 10:49:08 -05:00
bc4ef4861b
Default optimizer SGD
Alex Stoken
2020-06-24 10:07:43 -05:00
7abf202cad
Mode all optimizer settings to 'hyp.yaml', integrate proper momentum with Adam optimizer
Alex Stoken
2020-06-24 10:03:21 -05:00
7edbf6570e
Fix help message for cfg files
Alex Stoken
2020-06-24 09:45:57 -05:00
d64ad0fbf3
Remove --resume functionality and related checks/logic.
Alex Stoken
2020-06-24 09:17:27 -05:00
ccf0af18b1
Revert coco128.yaml to initial commit
Alex Stoken
2020-06-24 09:13:41 -05:00
f9ae460eec
update README.md
Glenn Jocher
2020-06-23 20:12:23 -07:00
ec72eea62b
Merge remote-tracking branch 'origin/master'
Glenn Jocher
2020-06-23 10:53:48 -07:00
53294f28b4
update yolo.py
Glenn Jocher
2020-06-23 10:53:42 -07:00
6cea1bff60
Update README.md
Glenn Jocher
2020-06-23 10:11:34 -07:00
a63e1c9b3c
study update
Glenn Jocher
2020-06-22 23:23:50 -07:00
2a550333c0
update README.md
Glenn Jocher
2020-06-22 23:20:37 -07:00
db898131f9
study update
Glenn Jocher
2020-06-22 23:07:22 -07:00
6c1b87a42e
update google_utils import
Glenn Jocher
2020-06-22 23:00:23 -07:00
7306dadf78
Update README.md
Glenn Jocher
2020-06-22 15:16:10 -07:00
3804578b78
Update README.md
Glenn Jocher
2020-06-22 14:33:47 -07:00
69ff486693
Update README.md
Glenn Jocher
2020-06-22 14:32:05 -07:00
364fcfd7db
PANet update
Glenn Jocher
2020-06-22 14:27:17 -07:00
ef58dac963
update Dockerfile
Glenn Jocher
2020-06-22 10:24:56 -07:00
bbd12c7cfe
detect.py fast bug fix
Glenn Jocher
2020-06-22 10:14:58 -07:00
fc171e2678
check_anchor_order() update
Glenn Jocher
2020-06-21 14:08:59 -07:00
1f1917ef56
remove fast, add merge
Glenn Jocher
2020-06-21 13:37:11 -07:00
e572bb0803
Add plot_results save location to log_dir
Alex Stoken
2020-06-21 09:36:28 -05:00
24dd150fbd
update torch>=1.4
Glenn Jocher
2020-06-20 11:55:39 -07:00
ba340ed72d
Merge pull request #143 from Lornatang/Fix-multi-GPU-training-errors
Glenn Jocher
2020-06-20 11:10:40 -07:00
024a42f4fd
datasets.py vid_formats updated
Glenn Jocher
2020-06-20 09:53:37 -07:00
5c2cd711df
vid_formats bug fix #146
Glenn Jocher
2020-06-20 09:26:07 -07:00
899f1d4bde
Fix DDP bug in single process multiple device use cases
Lornatang
2020-06-20 13:00:03 +08:00
0a09882e4f
Update issue templates
Glenn Jocher
2020-06-19 16:46:21 -07:00
cdf1eac9f7
Merge pull request #107 from Lornatang/fix-reference-bugs
Glenn Jocher
2020-06-19 14:51:41 -07:00
458c4d0e0b
Update README.md
Glenn Jocher
2020-06-19 12:14:06 -07:00
34ec87ecf9
Update README.md
Glenn Jocher
2020-06-19 12:13:15 -07:00
d4c6674c98
speeds updated from FP32 to FP16
Glenn Jocher
2020-06-19 12:05:10 -07:00
e249792c2f
update onnx_export.py usage
Glenn Jocher
2020-06-19 10:15:08 -07:00
80b82e8bde
update yolo.py
Glenn Jocher
2020-06-18 19:33:02 -07:00
245bd1d242
update README.md
Glenn Jocher
2020-06-18 12:46:30 -07:00
a3ee0a8954
check_img_size() update
Glenn Jocher
2020-06-18 11:46:04 -07:00
10145833c3
strip_optimizer() default to save FP16 model
Glenn Jocher
2020-06-18 00:16:47 -07:00
cce95e744d
backbone as FP16, save default to FP32
Glenn Jocher
2020-06-18 00:13:18 -07:00
d9b64c27c2
save ckpt in FP16 #119
Glenn Jocher
2020-06-17 22:34:13 -07:00
9fdb0fbacf
AutoAnchor bug fix # 117
Glenn Jocher
2020-06-17 19:51:15 -07:00
c8152c81a6
Syntax fixes
Alex Stoken
2020-06-17 16:32:13 -05:00
9b7386f603
Add save_dir arg to test.test, use arg as location for saving batch jpgs
Alex Stoken
2020-06-17 16:08:46 -05:00
945307beba
Add save_dir to plot_lr_scheduler and plot_labels Set save_dir = log_dir in train.py
Alex Stoken
2020-06-17 16:03:18 -05:00
3b2b330872
Move results.txt from weights/ to log_dir
Alex Stoken
2020-06-17 15:55:45 -05:00
8b6f5826bc
update README.md
Glenn Jocher
2020-06-17 12:58:01 -07:00
2bfd242fa1
Merge remote-tracking branch 'origin/master'
Glenn Jocher
2020-06-17 12:57:03 -07:00
b7ac446d62
update README.md
Glenn Jocher
2020-06-17 12:56:47 -07:00
ade023cff2
Fix hyp file read in and dict update. Add example of hyp yaml
Alex Stoken
2020-06-17 10:59:20 -05:00
3263a204ea
Fix get_latest_run to search 'search_dir' recursivly
Alex Stoken
2020-06-17 10:34:37 -05:00
d34291733b
Fix get_latest_run() to search inside 'weights' subfolders
Alex Stoken
2020-06-17 10:26:55 -05:00
91de3381e1
Merge pull request #112 from imprashr/patch-1
Glenn Jocher
2020-06-17 08:23:35 -07:00
0071b69994
onnx should be 1.6.0
Prashant Rai
2020-06-17 14:15:27 +05:30
2368603484
fix refrence bug
Lornatang
2020-06-17 09:56:26 +08:00
a557b7d924
check_img_size() update
Glenn Jocher
2020-06-16 17:48:01 -07:00
034609414e
AutoAnchor fitness to screen
Glenn Jocher
2020-06-16 17:08:45 -07:00
ef75f7313d
AutoAnchor n, imgsz bug fix
Glenn Jocher
2020-06-16 16:48:01 -07:00
2daa412575
train with multi-gpu half test bug fix #99
Glenn Jocher
2020-06-16 16:28:25 -07:00
d1e5716610
train with multi-gpu half test bug fix #99
Glenn Jocher
2020-06-16 16:22:58 -07:00
1dfc28527f
AutoAnchor BPR to 4 significant figures
Glenn Jocher
2020-06-16 16:19:58 -07:00
8db51c7002
tb_writer bug fix
Glenn Jocher
2020-06-16 16:05:28 -07:00
4052603e1f
AutoAnchor update - improvement check
Glenn Jocher
2020-06-16 15:54:20 -07:00
5f2eeba233
remove old print statements
Alex Stoken
2020-06-16 17:09:39 -05:00
afe1df385b
dist.destroy_process_group() bug fix
Glenn Jocher
2020-06-16 15:08:14 -07:00
72d06147b2
Multi-GPU disabled in test.py
Glenn Jocher
2020-06-16 14:42:30 -07:00
333f678b37
add update default hyp dict with provided yaml
Alex Stoken
2020-06-16 16:36:20 -05:00
a448c3bcd7
add logic for resuming and getting hyp for resume run
Alex Stoken
2020-06-16 16:30:12 -05:00
bafbc65ee3
AutoAnchor bug fix
Glenn Jocher
2020-06-16 14:26:02 -07:00
25e51bcec7
add util function to get most recent last.pt file added logic in train.py __main__ to handle resuming from a run
Alex Stoken
2020-06-16 15:50:27 -05:00
490f1e7b9c
add save_dir arg to plot_lr_scheduler, default to current dir. Uncomment plot_lr_scheduler in train() and pass log_dir as save location
Alex Stoken
2020-06-16 15:13:03 -05:00
57a0ae3350
AutoAnchor implementation
Glenn Jocher
2020-06-16 13:12:22 -07:00
4418809cf5
change weights dir (wdir) to be unique to each run, under log_dir
Alex Stoken
2020-06-16 15:09:51 -05:00
d9f446cd81
add save yaml of opt and hyp to tensorboard log_dir in train()
Alex Stoken
2020-06-16 15:06:13 -05:00
a85e6d0fc0
add parser arg for hyp yaml file
Alex Stoken
2020-06-16 14:53:32 -05:00
95c46f7245
Merge pull request #79 from Lornatang/delete-redundant-thrid-party-library
Glenn Jocher
2020-06-16 11:52:36 -07:00
5a50491fa1
check_anchors bug fix
Glenn Jocher
2020-06-16 10:36:35 -07:00
05b8ee5ca4
check_anchors() bug fix #102
Glenn Jocher
2020-06-16 10:34:16 -07:00
7c6fd4d670
--verbose bug fix #84
Glenn Jocher
2020-06-16 10:22:22 -07:00
ec81c7b5f2
check_anchors() bug fix #90
Glenn Jocher
2020-06-16 10:14:04 -07:00
e1e3399286
ONNX export bug fix #93
Glenn Jocher
2020-06-16 09:59:42 -07:00
8b26e89006
AutoAnchor bug fix #72
Glenn Jocher
2020-06-16 00:53:34 -07:00
8fa3724072
kmean_anchors() update
Glenn Jocher
2020-06-16 00:40:21 -07:00
bdd9fee841
update fast mode
Glenn Jocher
2020-06-16 00:29:54 -07:00
1c0b6236e3
update fast mode
Glenn Jocher
2020-06-16 00:11:29 -07:00
8b8b792802
There is no need to download extra packages, official bring it with you
Lornatang
2020-06-16 08:28:52 +08:00