Glenn Jocher
d68179953d
Update export.py
4年前
Glenn Jocher
1e7674d4dd
Update export.py --img-size comment
4年前
Glenn Jocher
4fb8cb353f
Update export.py with v3.0 Hardswish() support ( #831 )
4年前
Glenn Jocher
4d7f222f73
Update export.py with v3.0 Hardswish() support
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
2268f9c44f
update *.yaml output layer comments
4年前
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>
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
a97c3f94ec
update common.py Classify()
4年前
Glenn Jocher
5387d4747d
update common.py add Classify()
4年前
Glenn Jocher
2b1892430f
yolo.py bug fix
4年前
Glenn Jocher
02445d176d
improved model.yaml source tracking
4年前
Glenn Jocher
9006b85d21
update export.py torchscript file extension
4年前
Glenn Jocher
16f6834486
update train.py and experimental.py
4年前
Laurentiu Diaconu
a9918fbf2b
updated coreml conversion to do pixel scaling between 0.0 and 1.0
4年前
Glenn Jocher
e8cf24b6c8
Initial model ensemble capability #318
4年前
Glenn Jocher
04bdbe4104
fuse update
4年前
Glenn Jocher
5ba1de0cdc
update experimental.py with Ensemble() module
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
659ad748c8
update get_voc.sh
4年前
Glenn Jocher
7bd7b2c090
update export.py
4年前
Glenn Jocher
655895a838
module updates
4年前
Glenn Jocher
a9d20eba3e
export.py update
4年前
Glenn Jocher
569fbc63c8
export.py update
4年前
Glenn Jocher
1fca7a7f24
autopad() update in common.py
4年前
Glenn Jocher
a62a1c2c67
export.py update
4年前
Glenn Jocher
f1d67f4110
update export.py
4年前
Glenn Jocher
b5659d1195
module updates
4年前
Glenn Jocher
615d6d0cfa
FROM nvcr.io/nvidia/pytorch:20.06-py3
4年前
Glenn Jocher
0bc80e1e57
onnx_export.py bug fix #224
4年前
Jake Poznanski
685091a3fc
Adding torchscript export
4年前
Glenn Jocher
9a9c4f1259
update yolo.py default --cfg
4年前
Glenn Jocher
e071b8dd3a
update models/common.py for Conv() flexible padding
4年前
Glenn Jocher
0825cb7fd8
create models/hub
4年前
Glenn Jocher
b1b3634145
update onnx_export.py #134
4年前
Glenn Jocher
8699c319ca
update yolo.py
4年前
Glenn Jocher
53294f28b4
update yolo.py
4年前