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