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
hace 4 años
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>
hace 4 años
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>
hace 4 años
Jirka Borovec
d5b6416c87
Explicit Imports ( #498 )
* expand imports
* optimize
* miss
* fix
hace 4 años
Glenn Jocher
d989bc9260
remove NBSP
hace 4 años
Glenn Jocher
2268f9c44f
update *.yaml output layer comments
hace 4 años
Liu Changyu
adaf1d8ac1
Update models/hub *.yaml files for v2.0 ( #546 )
* fix #543 issue
* Update yolov3-spp.yaml
* Update yolov5-fpn.yaml
* Update yolov5-panet.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
hace 4 años
Glenn Jocher
43a616a955
PyTorch 1.6.0 compatability updates
hace 4 años
Glenn Jocher
91af0401d8
PyTorch 1.6.0 compatibility updates
hace 4 años
Glenn Jocher
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
hace 4 años
Glenn Jocher
9da56b62dd
v2.0 Release ( #491 )
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
hace 4 años
Glenn Jocher
a97c3f94ec
update common.py Classify()
hace 4 años
Glenn Jocher
5387d4747d
update common.py add Classify()
hace 4 años
Glenn Jocher
2b1892430f
yolo.py bug fix
hace 4 años
Glenn Jocher
02445d176d
improved model.yaml source tracking
hace 4 años
Glenn Jocher
9006b85d21
update export.py torchscript file extension
hace 4 años
Glenn Jocher
16f6834486
update train.py and experimental.py
hace 4 años
Laurentiu Diaconu
a9918fbf2b
updated coreml conversion to do pixel scaling between 0.0 and 1.0
hace 4 años
Glenn Jocher
e8cf24b6c8
Initial model ensemble capability #318
hace 4 años
Glenn Jocher
04bdbe4104
fuse update
hace 4 años
Glenn Jocher
5ba1de0cdc
update experimental.py with Ensemble() module
hace 4 años
Glenn Jocher
38f5c1ad1d
pruning and sparsity initial commit
hace 4 años
Glenn Jocher
997ba7b346
import yaml in yolo.py
hace 4 años
Glenn Jocher
4e2d24602d
update yolo.py
hace 4 años
Glenn Jocher
bb3c346916
model.yaml nc inherited from dataset.yaml
hace 4 años
Glenn Jocher
659ad748c8
update get_voc.sh
hace 4 años
Glenn Jocher
7bd7b2c090
update export.py
hace 4 años
Glenn Jocher
655895a838
module updates
hace 4 años
Glenn Jocher
a9d20eba3e
export.py update
hace 4 años
Glenn Jocher
569fbc63c8
export.py update
hace 4 años
Glenn Jocher
1fca7a7f24
autopad() update in common.py
hace 4 años
Glenn Jocher
a62a1c2c67
export.py update
hace 4 años
Glenn Jocher
f1d67f4110
update export.py
hace 4 años
Glenn Jocher
b5659d1195
module updates
hace 4 años
Glenn Jocher
615d6d0cfa
FROM nvcr.io/nvidia/pytorch:20.06-py3
hace 4 años
Glenn Jocher
0bc80e1e57
onnx_export.py bug fix #224
hace 4 años
Jake Poznanski
685091a3fc
Adding torchscript export
hace 4 años
Glenn Jocher
9a9c4f1259
update yolo.py default --cfg
hace 4 años
Glenn Jocher
e071b8dd3a
update models/common.py for Conv() flexible padding
hace 4 años
Glenn Jocher
0825cb7fd8
create models/hub
hace 4 años
Glenn Jocher
b1b3634145
update onnx_export.py #134
hace 4 años
Glenn Jocher
8699c319ca
update yolo.py
hace 4 años
Glenn Jocher
53294f28b4
update yolo.py
hace 4 años
Glenn Jocher
364fcfd7db
PANet update
hace 4 años
Glenn Jocher
fc171e2678
check_anchor_order() update
hace 4 años
Glenn Jocher
e249792c2f
update onnx_export.py usage
hace 4 años
Glenn Jocher
80b82e8bde
update yolo.py
hace 4 años
Glenn Jocher
e1e3399286
ONNX export bug fix #93
hace 4 años
ashing
62eda5e9c2
add "__init__.py" file in <models> folder
hace 4 años
Glenn Jocher
4ba47b1b1b
model.yaml updates
hace 4 años