2f77cf33f6
.fuse() additional error checking
2020-09-20 12:04:20 -07:00
6062bb7dee
print self.yaml['nc'] override update
2020-09-11 17:44:11 -07:00
35dc935504
remove `nn.Conv2d` in default settings ( #949 )
2020-09-11 16:35:48 -07:00
a8751e50de
model.fuse() fix for export.py ( #827 )
2020-08-24 19:27:54 -07:00
d0f98c0095
Update yolo.py with model_info(verbose=False)
2020-08-21 16:20:45 -07:00
b42e8a531b
optimize imports
2020-08-20 21:17:40 -07:00
916d4aad9a
v3.0 Release ( #725 )
...
* initial commit
* remove yolov3-spp from test.py study
* update study --img range
* update mAP
* cleanup and speed updates
* update README plot
2020-08-13 14:25:05 -07:00
0892c44bc4
Fix Logging ( #719 )
...
* Add logging setup
* Fix fusing layers message
* Fix logging does not have end
* Add logging
* Change logging to use logger
* Update yolo.py
I tried this in a cloned branch, and everything seems to work fine
* Update yolo.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-12 14:18:19 -07:00
4949401a94
Fix redundant outputs via Logging in DDP training ( #500 )
...
* Change print to logging
* Clean function set_logging
* Add line spacing
* Change leftover prints to log
* Fix scanning labels output
* Fix rank naming
* Change leftover print to logging
* Reorganized DDP variables
* Fix type error
* Make quotes consistent
* Fix spelling
* Clean function call
* Add line spacing
* Update datasets.py
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-11 11:18:45 -07:00
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
2020-08-02 15:47:36 -07:00
d989bc9260
remove NBSP
2020-08-02 14:23:05 -07:00
43a616a955
PyTorch 1.6.0 compatability updates
2020-07-28 15:38:56 -07:00
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
...
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
2020-07-24 11:42:23 -07:00
9da56b62dd
v2.0 Release ( #491 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-23 15:34:23 -07:00
2b1892430f
yolo.py bug fix
2020-07-13 15:21:19 -07:00
02445d176d
improved model.yaml source tracking
2020-07-13 14:35:47 -07:00
04bdbe4104
fuse update
2020-07-05 23:16:50 -07:00
38f5c1ad1d
pruning and sparsity initial commit
2020-07-05 13:41:21 -07:00
997ba7b346
import yaml in yolo.py
2020-07-05 12:50:04 -07:00
762b06de4b
Merge remote-tracking branch 'origin/master'
2020-07-04 23:07:32 -07:00
4e2d24602d
update yolo.py
2020-07-04 23:07:26 -07:00
bb3c346916
model.yaml nc inherited from dataset.yaml
2020-07-04 17:51:54 -07:00
655895a838
module updates
2020-07-03 13:46:12 -07:00
9a9c4f1259
update yolo.py default --cfg
2020-06-24 18:00:43 -07:00
8699c319ca
update yolo.py
2020-06-24 10:42:01 -07:00
53294f28b4
update yolo.py
2020-06-23 10:53:42 -07:00
fc171e2678
check_anchor_order() update
2020-06-21 14:08:59 -07:00
80b82e8bde
update yolo.py
2020-06-18 19:33:02 -07:00
8f17a96ff0
cleanup
2020-06-11 21:55:22 -07:00
8f1755290c
ONNX export update
2020-06-11 21:54:01 -07:00
a814720403
PyTorch Hub updates
2020-06-10 19:11:11 -07:00
c14368d768
new CSP model release
2020-06-09 14:31:04 -07:00
854ccb9a74
_print_weights()
2020-06-09 10:51:47 -07:00
5bee686649
model definition update
2020-06-08 23:26:56 -07:00
715cb08b10
module updates
2020-06-08 22:15:16 -07:00
ff02ae0869
module updates
2020-06-08 22:13:01 -07:00
12b0c046d5
model fusion and onnx export
2020-06-07 13:42:33 -07:00
b810b21299
augmented inference
2020-06-05 21:14:15 -07:00
5c470d235e
multi-gpu test bug fix #15
2020-06-05 20:36:29 -07:00
67d5e65c77
profiling update
2020-06-05 20:22:39 -07:00
2703ac73e8
profiling update
2020-06-05 20:22:17 -07:00
dbdee3a4a3
multi-gpu test bug fix #15
2020-06-04 15:23:03 -07:00
a2336088f0
updates
2020-06-04 14:25:10 -07:00
ee8988b8a2
datasets.py self.batch_shapes fix
2020-06-01 18:11:18 -07:00
f129cdda9f
updates
2020-05-30 11:03:50 -07:00
1b97392801
updates
2020-05-30 10:46:35 -07:00
1e84a23f38
initial commit
2020-05-29 17:04:54 -07:00
Glenn Jocher
tomguluson92
NanoCode012
Jirka Borovec