Glenn Jocher
d8f5fcfe87
Improved FLOPS computation ( #1398 )
* Improved FLOPS computation
* update comment
4 years ago
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4 years ago
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup ( #1325 )
* Create flatten_recursive() helper function
* cleanup
* print torch version
4 years ago
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
* torch 1.7.0 compatibility updates
* add inference verification
4 years ago
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
4 years ago
Glenn Jocher
c2403d785c
fuse() bug fix
4 years ago
Glenn Jocher
a12698f4c9
Update ci-testing.yml reflecting sys.path additions ( #869 )
* Update ci-testing.yml
* Update yolo.py
* Update ci-testing.yml
* Update export.py
* Update yolo.py
4 years ago
Glenn Jocher
4fce0096fa
model.add_nms() method
4 years ago
Glenn Jocher
2f77cf33f6
.fuse() additional error checking
4 years ago
Glenn Jocher
6062bb7dee
print self.yaml['nc'] override update
4 years ago
tomguluson92
35dc935504
remove `nn.Conv2d` in default settings ( #949 )
4 years ago
Glenn Jocher
a8751e50de
model.fuse() fix for export.py ( #827 )
4 years ago
Glenn Jocher
d0f98c0095
Update yolo.py with model_info(verbose=False)
4 years ago
Glenn Jocher
b42e8a531b
optimize imports
4 years ago
Glenn Jocher
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
4 years ago
NanoCode012
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>
4 years ago
NanoCode012
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>
4 years ago
Jirka Borovec
d5b6416c87
Explicit Imports ( #498 )
* expand imports
* optimize
* miss
* fix
4 years ago
Glenn Jocher
d989bc9260
remove NBSP
4 years ago
Glenn Jocher
43a616a955
PyTorch 1.6.0 compatability updates
4 years ago
Glenn Jocher
91af0401d8
PyTorch 1.6.0 compatibility updates
4 years ago
Glenn Jocher
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
4 years ago
Glenn Jocher
9da56b62dd
v2.0 Release ( #491 )
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
Glenn Jocher
2b1892430f
yolo.py bug fix
4 years ago
Glenn Jocher
02445d176d
improved model.yaml source tracking
4 years ago
Glenn Jocher
04bdbe4104
fuse update
4 years ago
Glenn Jocher
38f5c1ad1d
pruning and sparsity initial commit
4 years ago
Glenn Jocher
997ba7b346
import yaml in yolo.py
4 years ago
Glenn Jocher
4e2d24602d
update yolo.py
4 years ago
Glenn Jocher
bb3c346916
model.yaml nc inherited from dataset.yaml
4 years ago
Glenn Jocher
655895a838
module updates
4 years ago
Glenn Jocher
9a9c4f1259
update yolo.py default --cfg
4 years ago
Glenn Jocher
8699c319ca
update yolo.py
4 years ago
Glenn Jocher
53294f28b4
update yolo.py
4 years ago
Glenn Jocher
fc171e2678
check_anchor_order() update
4 years ago
Glenn Jocher
80b82e8bde
update yolo.py
4 years ago
Glenn Jocher
8f17a96ff0
cleanup
4 years ago
Glenn Jocher
8f1755290c
ONNX export update
4 years ago
Glenn Jocher
a814720403
PyTorch Hub updates
4 years ago
Glenn Jocher
c14368d768
new CSP model release
4 years ago
Glenn Jocher
854ccb9a74
_print_weights()
4 years ago
Glenn Jocher
5bee686649
model definition update
4 years ago
Glenn Jocher
715cb08b10
module updates
4 years ago
Glenn Jocher
ff02ae0869
module updates
4 years ago
Glenn Jocher
12b0c046d5
model fusion and onnx export
4 years ago
Glenn Jocher
b810b21299
augmented inference
4 years ago
Glenn Jocher
5c470d235e
multi-gpu test bug fix #15
4 years ago
Glenn Jocher
67d5e65c77
profiling update
4 years ago
Glenn Jocher
2703ac73e8
profiling update
4 years ago
Glenn Jocher
dbdee3a4a3
multi-gpu test bug fix #15
4 years ago