Glenn Jocher
f5429260ca
PyTorch Hub and autoShape update ( #1415 )
* PyTorch Hub and autoShape update
* comment x for imgs
* reduce comment
4年前
Glenn Jocher
d8f5fcfe87
Improved FLOPS computation ( #1398 )
* Improved FLOPS computation
* update comment
4年前
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4年前
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup ( #1325 )
* Create flatten_recursive() helper function
* cleanup
* print torch version
4年前
Glenn Jocher
15a10609fe
autoshape() update for PIL greyscale inputs ( #1279 )
* autoshape update for PIL greyscale inputs
* autoshape update for PIL greyscale inputs
4年前
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
* torch 1.7.0 compatibility updates
* add inference verification
4年前
Yi_Zhang
f532bc68a9
GhostConv() bug fix ( #1176 )
4年前
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
4年前
Glenn Jocher
d45e349a1e
Rearrange export input after checks ( #1118 )
img size checks are warnings rather than errors, so current implementation allows improperly formed model inputs.
4年前
Glenn Jocher
c2403d785c
fuse() bug fix
4年前
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
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年前
PetrDvoracek
d55c326dda
export.py folder sys.path addition ( #1052 )
4年前
Glenn Jocher
4fce0096fa
model.add_nms() method
4年前
Glenn Jocher
2f77cf33f6
.fuse() additional error checking
4年前
Glenn Jocher
c4cb78570c
add NMS to pretrained pytorch hub models
4年前
Glenn Jocher
6062bb7dee
print self.yaml['nc'] override update
4年前
tomguluson92
35dc935504
remove `nn.Conv2d` in default settings ( #949 )
4年前
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年前