2478 Commits

Author SHA1 Message Date
Glenn Jocher
316d99c377 yolov5 regress updates to yolov3 2020-05-17 15:19:33 -07:00
Glenn Jocher
c8f4ee6c46 yolov5 regress updates to yolov3 - build_targets() 2020-05-17 15:10:31 -07:00
Glenn Jocher
110ead20e6 yolov5 regress updates to yolov3 2020-05-17 15:00:07 -07:00
Glenn Jocher
c94019f159 iglob bug fix 2020-05-17 14:31:14 -07:00
Glenn Jocher
bbd82bb94d updates 2020-05-17 14:30:12 -07:00
Glenn Jocher
6bfb3a96c8 iglob bug fix 2020-05-16 22:43:12 -07:00
Glenn Jocher
37bd5490ef iglob file-search improvements 2020-05-16 22:25:21 -07:00
Glenn Jocher
27c7b02fff --save-txt extension fix 2020-05-16 11:51:49 -07:00
orcund
3a71daf4bc
Pseudo Labeling (#1149)
* Added pseudo labeling

* Delete print_test.py

* Refactor label generation

* Update detect.py

* Update detect.py

* Update utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-05-16 11:09:57 -07:00
Glenn Jocher
3f27ef1253 pycocotools and numpy 1.17 fix for #1182 2020-05-15 20:50:58 -07:00
Glenn Jocher
f6a19d5b32 add cd53-based *.cfg 2020-05-15 16:20:04 -07:00
Glenn Jocher
20891926c9 add stride order reversal for c53*.cfg 2020-05-15 14:40:14 -07:00
Glenn Jocher
6fe67595cb add stride order reversal for c53*.cfg 2020-05-15 11:32:23 -07:00
IlyaOvodov
b2fcfc573e
convert(...) changed to save converted file alongside the original file (#1167) 2020-05-13 09:08:55 -07:00
Glenn Jocher
c066d7d439 Merge remote-tracking branch 'origin/master' 2020-05-12 09:53:18 -07:00
Glenn Jocher
0cf88f046d hyp evolution bug fix #1160 2020-05-12 09:53:13 -07:00
Glenn Jocher
031c2144ec
Update README.md 2020-05-12 08:31:36 -07:00
Glenn Jocher
894a3e54ca
Update --bug-report.md 2020-05-11 11:14:34 -07:00
Glenn Jocher
9f04e175f6 nms torch.mm() update 2020-05-10 11:26:37 -07:00
Glenn Jocher
ae2bc020eb git status check - linux and darwin 2020-05-09 22:35:44 -07:00
Glenn Jocher
965155ee60 CUBLAS bug fix #1139 2020-05-06 10:26:28 -07:00
Glenn Jocher
832ceba559 update bug report template 2020-05-06 10:14:31 -07:00
Glenn Jocher
d405959893 cleanup 2020-05-04 13:33:34 -07:00
Glenn Jocher
5d42cc1b9a
Update README.md 2020-05-02 18:44:31 -07:00
Glenn Jocher
b0b52eec53 yolov4 tensorrt 2020-05-02 11:09:09 -07:00
Glenn Jocher
23614b8c2e speed update 2020-05-02 10:24:26 -07:00
Glenn Jocher
add73a0e74 speed update 2020-05-02 10:23:40 -07:00
Glenn Jocher
ee7cba65a5 kmeans() cleanup 2020-05-02 09:20:19 -07:00
Glenn Jocher
be87b41aa2 update image display per #1114 2020-04-30 16:50:58 -07:00
Glenn Jocher
b0629d622c bug fix on #1114 2020-04-30 15:03:32 -07:00
Glenn Jocher
0ffbf5534e cleanup for #1114 2020-04-30 14:53:57 -07:00
Josh Veitch-Michaelis
fb1b5e09b2
faster and more informative training plots (#1114)
* faster and more informative training plots

* Update utils.py

Looks good. Needs pep8 linting, I'll do that in PyCharm later once PR is in.

* Update test.py

* Update train.py

f for the tb descriptor lets us plot several batches, i.e. to allow us to change L292 to 'if ni < 3' for 3 examples.

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-04-30 13:37:04 -07:00
Glenn Jocher
f1d73a29e5 Optimizer group report 2020-04-30 12:26:02 -07:00
Glenn Jocher
d62d68929c cleanup 2020-04-29 12:00:30 -07:00
Glenn Jocher
9f88f5cc21 cleanup 2020-04-29 11:34:59 -07:00
Glenn Jocher
9cc4951d4f auto reverse-strides for yolov4/panet 2020-04-28 15:24:14 -07:00
Glenn Jocher
c6ea2b58ea auto-accumulate update 2020-04-28 15:06:33 -07:00
Glenn Jocher
37cbe89ef0 test/train jpg for png 2020-04-28 13:45:27 -07:00
Josh Veitch-Michaelis
992d8af242
faster hsv augmentation (#1110)
As per https://github.com/ultralytics/yolov3/issues/1096
2020-04-28 12:59:44 -07:00
Glenn Jocher
15f1343dfc uncached label removal 2020-04-28 11:07:26 -07:00
Glenn Jocher
b1d385a8de yolov4-relu.cfg 2020-04-27 21:19:22 -07:00
Glenn Jocher
02ae0e3bbd reproduce results update 2020-04-27 21:05:19 -07:00
Glenn Jocher
8521c3cff9 cleanup 2020-04-27 15:22:36 -07:00
Glenn Jocher
e9d41bb566 Speed updated 2020-04-27 15:06:26 -07:00
Glenn Jocher
2518868508 MemoryEfficientMish() 2020-04-27 13:51:21 -07:00
Glenn Jocher
3aa347a321 add HardSwish() 2020-04-27 13:08:24 -07:00
Glenn Jocher
692f945819 Merge remote-tracking branch 'origin/master' 2020-04-27 11:20:33 -07:00
Glenn Jocher
f799c15611 result not updated from pycocotools 2020-04-27 11:20:27 -07:00
Josh Veitch-Michaelis
18702c9608
add tensorboard to requirements (#1100)
In a clean environment running training fails if tensorboard is not installed e.g.

```
Traceback (most recent call last):
  File "/home/josh/miniconda3/envs/ultralytics/lib/python3.7/site-packages/torch/utils/tensorboard/__init__.py", line 2, in <module>
    from tensorboard.summary.writer.record_writer import RecordWriter  # noqa F401
ModuleNotFoundError: No module named 'tensorboard'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "train.py", line 6, in <module>
    from torch.utils.tensorboard import SummaryWriter
  File "/home/josh/miniconda3/envs/ultralytics/lib/python3.7/site-packages/torch/utils/tensorboard/__init__.py", line 4, in <module>
    raise ImportError('TensorBoard logging requires TensorBoard with Python summary writer installed. '
ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed. This should be available in 1.14 or above.
```
2020-04-27 09:21:34 -07:00
Glenn Jocher
4a4bfb20de FLOPS verbose=False 2020-04-26 16:31:57 -07:00