mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-02 22:29:06 +08:00
Code Issues Projects Releases Wiki Activity
2,799 Commits 10 Branches 10 Tags
Commit Graph

2799 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
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
Alex Stoken
5f2eeba233 remove old print statements 2020-06-16 17:09:39 -05: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
Alex Stoken
333f678b37 add update default hyp dict with provided yaml 2020-06-16 16:36:20 -05:00
Alex Stoken
a448c3bcd7 add logic for resuming and getting hyp for resume run 2020-06-16 16:30:12 -05:00
Glenn Jocher
bafbc65ee3 AutoAnchor bug fix 2020-06-16 14:26:02 -07:00
Alex Stoken
25e51bcec7 add util function to get most recent last.pt file 
added logic in train.py __main__ to handle resuming from a run
 2020-06-16 15:50:27 -05:00
Alex Stoken
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
 2020-06-16 15:13:03 -05:00
Glenn Jocher
57a0ae3350 AutoAnchor implementation 2020-06-16 13:12:22 -07:00
Alex Stoken
4418809cf5 change weights dir (wdir) to be unique to each run, under log_dir 2020-06-16 15:09:51 -05:00
Alex Stoken
d9f446cd81 add save yaml of opt and hyp to tensorboard log_dir in train() 2020-06-16 15:06:13 -05:00
Alex Stoken
a85e6d0fc0 add parser arg for hyp yaml file 2020-06-16 14:53:32 -05: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
Glenn Jocher
14523bb030 FP16 to FP32 ckpt load 2020-06-15 13:18:39 -07:00
Glenn Jocher
c5966abba8 glob search bug fix #77 2020-06-15 12:08:57 -07:00
Glenn Jocher
ace56eb5b6
Merge pull request #70 from Lornatang/fix-reference-bugs 
fix reference problem
 2020-06-15 11:18:17 -07:00
Lornatang
54795a4f0a fix problem 2020-06-15 16:11:06 +08:00
Glenn Jocher
45a17ea40c Merge remote-tracking branch 'origin/master' 2020-06-15 00:44:08 -07:00
Glenn Jocher
e7c24e04ac half() fix #66 2020-06-15 00:43:57 -07:00
Glenn Jocher
6570336284
Merge pull request #65 from Ashing00/master 
add "__init__.py" file in <models> folder
 2020-06-15 00:34:54 -07:00