Commit Graph

2473 Commits

Author SHA1 Message Date
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
Glenn Jocher a0a3bab9e6 add Mish() support 2020-04-26 16:31:21 -07:00
Glenn Jocher 18d4ebfd12 add Mish() support 2020-04-26 16:25:46 -07:00
Glenn Jocher 11f228eb00 yolov4.cfg from alexeyab/darknet 2020-04-26 16:07:29 -07:00
Glenn Jocher 55757421de remove future 2020-04-26 14:09:12 -07:00
Glenn Jocher efbeb283c4 ONNX grid float 2020-04-26 14:01:20 -07:00