Commit Graph

1979 Commits (1a3ecb8b386115fd22129eaf0760157b161efac7)
 

Author SHA1 Message Date
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
ashing 62eda5e9c2 add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
Glenn Jocher 111994956e switch default inference to FP16 on GPU 2020-06-14 19:25:39 -07:00
Glenn Jocher 157aff2854 FP16 inference fix 2020-06-14 18:45:20 -07:00
Glenn Jocher c6fc4242a2 Merge remote-tracking branch 'origin/master' 2020-06-14 18:32:35 -07:00
Glenn Jocher 260b1729f0 FP16 inference update 2020-06-14 18:32:29 -07:00
Glenn Jocher b545ee32f4 Created using Colaboratory 2020-06-14 13:29:05 -07:00
Glenn Jocher 629d3701fc pyyaml >= 5.3 2020-06-14 12:42:51 -07:00
Glenn Jocher 4b0e3745f2
Merge pull request #59 from Lijun-Yu/master
Fix config path for PyTorch Hub
2020-06-14 12:16:41 -07:00
Glenn Jocher 4f6c0cf4c5
Update hubconf.py 2020-06-14 12:15:14 -07:00
Lijun Yu 0dec8ffd66 path fix for torch hub 2020-06-14 14:47:40 -04:00
Glenn Jocher 4dd75b3635 pycocotools git install fix 2020-06-14 10:54:00 -07:00
Glenn Jocher 9bf35e7b86
Merge pull request #56 from channeladam/master
Fix initial basic setup / new starter issues
2020-06-14 10:33:34 -07:00
Glenn Jocher 335c607a24
Update datasets.py 2020-06-14 10:32:59 -07:00
Glenn Jocher 1374581929
Update requirements.txt 2020-06-14 10:23:33 -07:00
Adam f47ca2085c Include yolov5x extra large model weights 2020-06-14 18:19:23 +10:00
Adam 34bf936ce6 Add more logging for diagnostics 2020-06-14 18:18:57 +10:00
Adam faa6aeaa20 Fix pip requirements issues 2020-06-14 18:18:31 +10:00
Glenn Jocher 93bbf1b87a best possible recall fix 2020-06-13 15:25:53 -07:00
Glenn Jocher 31f3310029 assert best possible recall > 0.9 before training 2020-06-13 15:05:41 -07:00
Glenn Jocher 19e68e8a7b
Merge pull request #51 from KPLauritzen/b-ensure-weights-dir-when-training
Ensures weights/ dir exists
2020-06-13 12:54:43 -07:00