Glenn Jocher
0ff5aeca61
Create date_modified() ( #2616 )
Updated device selection string with fallback for non-git directories.
```python
def select_device(device='', batch_size=None):
# device = 'cpu' or '0' or '0,1,2,3'
s = f'YOLOv5 🚀 {git_describe() or date_modified()} torch {torch.__version__} ' # string
...
```
pirms 3 gadiem
Glenn Jocher
77415a42e5
Update git_describe() for remote dir usage ( #2606 )
pirms 3 gadiem
Glenn Jocher
9b11f0c58b
PyTorch Hub models default to CUDA:0 if available ( #2472 )
* PyTorch Hub models default to CUDA:0 if available
* device as string bug fix
pirms 3 gadiem
Glenn Jocher
f01f3223d5
Integer printout ( #2450 )
* Integer printout
* test.py 'Labels'
* Update train.py
pirms 3 gadiem
xiaowo1996
83dc1b4484
FLOPS min stride 32 ( #2276 )
Signed-off-by: xiaowo1996 <429740343@qq.com>
pirms 3 gadiem
Glenn Jocher
ffef77124e
Update torch_utils.py ( #1895 )
pirms 3 gadiem
Glenn Jocher
6ab589583c
Add colorstr() ( #1887 )
* Add colorful()
* update
* newline fix
* add git description
* --always
* update loss scaling
* update loss scaling 2
* rename to colorstr()
pirms 3 gadiem
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
pirms 3 gadiem
Glenn Jocher
9f5a18bb80
Torch CUDA synchronize update ( #1826 )
* torch.cuda.synchronize() update
* torch.cuda.synchronize() update
* torch.cuda.synchronize() update
* newline
pirms 3 gadiem
Glenn Jocher
c0ffcdf998
Display correct CUDA devices ( #1776 )
* Display correct CUDA devices
* cleanup
pirms 3 gadiem
Glenn Jocher
3004fb5bc1
Automatic m.half() profile on x.half()
pirms 3 gadiem
Glenn Jocher
0bd9c48609
Update torch_utils.py
FLOPS to GFLOPS
pirms 3 gadiem
Glenn Jocher
394d1c89f3
Input channel yaml['ch'] addition ( #1741 )
pirms 3 gadiem
Glenn Jocher
8bc0027afc
Update loss criteria constructor ( #1711 )
pirms 3 gadiem
NanoCode012
035ac82ed0
Fix torch multi-GPU --device error ( #1701 )
* Fix torch GPU error
* Update torch_utils.py
single-line device =
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pirms 3 gadiem
Glenn Jocher
ada90e3901
Profile() feature addition ( #1673 )
* Profile() feature addition
* cleanup
pirms 3 gadiem
Glenn Jocher
8918e63476
Increase FLOPS robustness ( #1608 )
pirms 3 gadiem
yujun
05a955a3f6
FLOPS computation device bug fix ( #1447 )
* Update torch_utils.py
fix issue#113 , inputs device should be same with model parameters' device
* Update torch_utils.py
* Update torch_utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pirms 4 gadiem
Glenn Jocher
9f27902368
Improved FLOPS computation ( #1398 )
pirms 4 gadiem
Glenn Jocher
d8f5fcfe87
Improved FLOPS computation ( #1398 )
* Improved FLOPS computation
* update comment
pirms 4 gadiem
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
pirms 4 gadiem
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup ( #1325 )
* Create flatten_recursive() helper function
* cleanup
* print torch version
pirms 4 gadiem
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
* torch 1.7.0 compatibility updates
* add inference verification
pirms 4 gadiem
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
pirms 4 gadiem
Glenn Jocher
124f0e8212
torchvision nms bug fix
pirms 4 gadiem
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
pirms 4 gadiem
Glenn Jocher
89655a84f2
.fuse() gradient introduction bug fix
pirms 4 gadiem
tkianai
c5969f79f5
add support for DWConv fuse operation ( #898 )
* the value should be a list to unpacking
* filter the None labels
* update readme to remove apex dependency
* add support for DWConv fuse operation
pirms 4 gadiem
Glenn Jocher
b42e8a531b
optimize imports
pirms 4 gadiem
Francisco Ingham
26c3b11fdc
Update torch_utils.py ( #751 )
pirms 4 gadiem
Glenn Jocher
f5da528d28
reformat code
pirms 4 gadiem
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>
pirms 4 gadiem
Glenn Jocher
3c6e2f7668
Single-source training ( #680 )
* Single-source training
* Extract hyperparameters into seperate files
* weight decay scientific notation yaml reader bug fix
* remove import glob
* intersect_dicts() implementation
* 'or' bug fix
* .to(device) bug fix
pirms 4 gadiem
Liu Changyu
c020875b17
PyTorch 1.6.0 update with native AMP ( #573 )
* PyTorch have Automatic Mixed Precision (AMP) Training.
* Fixed the problem of inconsistent code length indentation
* Fixed the problem of inconsistent code length indentation
* Mixed precision training is turned on by default
pirms 4 gadiem
Glenn Jocher
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
pirms 4 gadiem
Glenn Jocher
9da56b62dd
v2.0 Release ( #491 )
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pirms 4 gadiem
Glenn Jocher
2377e5f6ee
FP16 EMA bug fix
pirms 4 gadiem
Glenn Jocher
41ff83a98d
update torch_utils.py to FP16 EMA
pirms 4 gadiem
Glenn Jocher
a586751904
multi-gpu ckpt filesize bug fix #253
pirms 4 gadiem
Glenn Jocher
24c5a941f0
--resume EMA fix #292
pirms 4 gadiem
Glenn Jocher
121d90b3f2
update fuse_conv_and_bn()
pirms 4 gadiem
Glenn Jocher
38f5c1ad1d
pruning and sparsity initial commit
pirms 4 gadiem
NanoCode012
ccfa9373ae
Update to remove "module" attribute check
pirms 4 gadiem
NanoCode012
f83521ab9a
Fix save error for multi-gpu
pirms 4 gadiem
Glenn Jocher
af41083d52
EMA FP16 fix #279
pirms 4 gadiem
Glenn Jocher
df224a0d8f
EMA bug fix #279
pirms 4 gadiem
Glenn Jocher
f02481c73a
Update torch_utils.py
pirms 4 gadiem
Lornatang
8b8b792802
There is no need to download extra packages, official bring it with you
I submitted it once in your yolov3 project, you seem to accept it? I'm not sure. I'll submit PR again.
pirms 4 gadiem
Glenn Jocher
8fe299f179
model fuse
pirms 4 gadiem
Glenn Jocher
b810b21299
augmented inference
pirms 4 gadiem