Abhiram V
4501169547
Update yolo.py with yaml.SafeLoader ( #1970 )
3 years ago
lorne
03ebe6ea14
Add nn.SiLU inplace in attempt_load() ( #1940 )
* sleep 3 sec to load data .
* Update train.py
* Add nn.SiLU inplace in attempt_load()
Co-authored-by: wangliang <wangliang@codingapi.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
1d1c0567a4
PyTorch Hub results.render() ( #1897 )
3 years ago
Glenn Jocher
69be8e738f
YOLOv5 v4.0 Release ( #1837 )
* Update C3 module
* Update C3 module
* Update C3 module
* Update C3 module
* update
* update
* update
* update
* update
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update datasets
* update
* update
* update
* update attempt_downlaod()
* merge
* merge
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* parameterize eps
* comments
* gs-multiple
* update
* max_nms implemented
* Create one_cycle() function
* update
* update
* update
* update
* update
* update
* update
* update study.png
* update study.png
* Update datasets.py
3 years ago
Glenn Jocher
0b6266f5e0
Update yolov3-tiny.yaml ( #1823 )
3 years ago
NanoCode012
14b0abe2d0
autoShape() default for PyTorch Hub models ( #1692 )
* Add autoshape parameter
* Remove autoshape call in ReadMe
* Update hubconf.py
* file/URI inputs and autoshape check passthrough
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
9922c1183f
leaf Variable inplace bug fix ( #1759 )
3 years ago
Glenn Jocher
394d1c89f3
Input channel yaml['ch'] addition ( #1741 )
3 years ago
Glenn Jocher
6bd5e8bca7
nn.SiLU() export support ( #1713 )
3 years ago
Glenn Jocher
799724108f
Update C3 module ( #1705 )
3 years ago
Glenn Jocher
fa8f1fb0e9
Simplify autoshape() post-process ( #1653 )
* Simplify autoshape() post-process
* cleanup
* cleanup
3 years ago
Glenn Jocher
791dadb51c
Pycocotools best.pt after COCO train ( #1616 )
* Pycocotools best.pt after COCO train
* cleanup
3 years ago
Glenn Jocher
d929bb656c
Implement default class names ( #1609 )
3 years ago
Glenn Jocher
ba48f867ea
Add bias to Classify() ( #1601 )
3 years ago
Glenn Jocher
199c9c7874
Detection() device bug fix ( #1455 )
4 years ago
Glenn Jocher
225845e781
Update test.py for IoU in native image-space ( #1439 )
* Update test.py for IoU in native image-space
* remove redundant
* gn to device
* remove output scale_coords
* --img-size correction
* update
* native-space labels
* pred to predn
* remove clip_coords()
4 years ago
Glenn Jocher
97c6a59dd1
PANet anchor order correction ( #1420 )
4 years ago
Glenn Jocher
f5429260ca
PyTorch Hub and autoShape update ( #1415 )
* PyTorch Hub and autoShape update
* comment x for imgs
* reduce comment
4 years ago
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
15a10609fe
autoshape() update for PIL greyscale inputs ( #1279 )
* autoshape update for PIL greyscale inputs
* autoshape update for PIL greyscale inputs
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
Yi_Zhang
f532bc68a9
GhostConv() bug fix ( #1176 )
4 years ago
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
4 years ago
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 years ago
Glenn Jocher
c2403d785c
fuse() bug fix
4 years ago
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
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
PetrDvoracek
d55c326dda
export.py folder sys.path addition ( #1052 )
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
c4cb78570c
add NMS to pretrained pytorch hub models
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
d68179953d
Update export.py
4 years ago
Glenn Jocher
1e7674d4dd
Update export.py --img-size comment
4 years ago
Glenn Jocher
4fb8cb353f
Update export.py with v3.0 Hardswish() support ( #831 )
4 years ago
Glenn Jocher
4d7f222f73
Update export.py with v3.0 Hardswish() support
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
2268f9c44f
update *.yaml output layer comments
4 years ago
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 years ago
Glenn Jocher
43a616a955
PyTorch 1.6.0 compatability updates
4 years ago