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
Glenn Jocher
3bf0cb9c60
remove tb-nightly
2020-04-26 12:41:15 -07:00
Glenn Jocher
daedfc5487
reduce merge limit to 3000
2020-04-26 11:22:29 -07:00