Glenn Jocher
|
33202b7f0b
|
YOLOv5 + Albumentations integration (#3882)
* Albumentations integration
* ToGray p=0.01
* print confirmation
* create instance in dataloader init method
* improved version handling
* transform not defined fix
* assert string update
* create check_version()
* add spaces
* update class comment
|
3年前 |
Glenn Jocher
|
ef0b5c9d29
|
On-demand `pycocotools` pip install (#3547)
|
3年前 |
Glenn Jocher
|
f3c3d2ce5d
|
Merge `develop` branch into `master` (#3518)
* update ci-testing.yml (#3322)
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
* update ci-testing.yml (#3322)
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
* Enable direct `--weights URL` definition (#3373)
* Enable direct `--weights URL` definition
@KalenMike this PR will enable direct --weights URL definition. Example use case:
```
python train.py --weights https://storage.googleapis.com/bucket/dir/model.pt
```
* cleanup
* bug fixes
* weights = attempt_download(weights)
* Update experimental.py
* Update hubconf.py
* return bug fix
* comment mirror
* min_bytes
* Update tutorial.ipynb (#3368)
add Open in Kaggle badge
* `cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379)
* Update datasets.py
* comment
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* COCO evolution fix (#3388)
* COCO evolution fix
* cleanup
* update print
* print fix
* Create `is_pip()` function (#3391)
Returns `True` if file is part of pip package. Useful for contextual behavior modification.
```python
def is_pip():
# Is file in a pip package?
return 'site-packages' in Path(__file__).absolute().parts
```
* Revert "`cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379)" (#3395)
This reverts commit 21a9607e00 .
* Update FLOPs description (#3422)
* Update README.md
* Changing FLOPS to FLOPs.
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
* Parse URL authentication (#3424)
* Parse URL authentication
* urllib.parse.unquote()
* improved error handling
* improved error handling
* remove %3F
* update check_file()
* Add FLOPs title to table (#3453)
* Suppress jit trace warning + graph once (#3454)
* Suppress jit trace warning + graph once
Suppress harmless jit trace warning on TensorBoard add_graph call. Also fix multiple add_graph() calls bug, now only on batch 0.
* Update train.py
* Update MixUp augmentation `alpha=beta=32.0` (#3455)
Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
* Add `timeout()` class (#3460)
* Add `timeout()` class
* rearrange order
* Faster HSV augmentation (#3462)
remove datatype conversion process that can be skipped
* Add `check_git_status()` 5 second timeout (#3464)
* Add check_git_status() 5 second timeout
This should prevent the SSH Git bug that we were discussing @KalenMike
* cleanup
* replace timeout with check_output built-in timeout
* Improved `check_requirements()` offline-handling (#3466)
Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
* Add `output_names` argument for ONNX export with dynamic axes (#3456)
* Add output names & dynamic axes for onnx export
Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3
* use first output only + cleanup
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Revert FP16 `test.py` and `detect.py` inference to FP32 default (#3423)
* fixed inference bug ,while use half precision
* replace --use-half with --half
* replace space and PEP8 in detect.py
* PEP8 detect.py
* update --half help comment
* Update test.py
* revert space
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Add additional links/resources to stale.yml message (#3467)
* Update stale.yml
* cleanup
* Update stale.yml
* reformat
* Update stale.yml HUB URL (#3468)
* Stale `github.actor` bug fix (#3483)
* Explicit `model.eval()` call `if opt.train=False` (#3475)
* call model.eval() when opt.train is False
call model.eval() when opt.train is False
* single-line if statement
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* check_requirements() exclude `opencv-python` (#3495)
Fix for 3rd party or contrib versions of installed OpenCV as in https://github.com/ultralytics/yolov5/issues/3494.
* Earlier `assert` for cpu and half option (#3508)
* early assert for cpu and half option
early assert for cpu and half option
* Modified comment
Modified comment
* Update tutorial.ipynb (#3510)
* Reduce test.py results spacing (#3511)
* Update README.md (#3512)
* Update README.md
Minor modifications
* 850 width
* Update greetings.yml
revert greeting change as PRs will now merge to master.
Co-authored-by: Piotr Skalski <SkalskiP@users.noreply.github.com>
Co-authored-by: SkalskiP <piotr.skalski92@gmail.com>
Co-authored-by: Peretz Cohen <pizzaz93@users.noreply.github.com>
Co-authored-by: tudoulei <34886368+tudoulei@users.noreply.github.com>
Co-authored-by: chocosaj <chocosaj@users.noreply.github.com>
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Sam_S <SamSamhuns@users.noreply.github.com>
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: edificewang <609552430@qq.com>
|
3年前 |
chocosaj
|
3cb9ad4fc4
|
Update FLOPs description (#3422)
* Update README.md
* Changing FLOPS to FLOPs.
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
|
3年前 |
Glenn Jocher
|
ee24ae11ea
|
Comment Cython (#3297)
Ultralytics REST API fix.
|
3年前 |
Glenn Jocher
|
7b863a34dc
|
Add Cython (#3217)
Cython required by pycocotools in certain cases, i.e. https://github.com/cocodataset/cocoapi/issues/172
|
3年前 |
Glenn Jocher
|
b098600763
|
Update requirements.txt `onnx>=1.9.0` (#3143)
|
3年前 |
Glenn Jocher
|
fca16dc4b3
|
Remove Cython from requirements.txt (#2604)
Cython should be a dependency of the remaining packages in requirements.txt, so should be installed anyway even if not a direct requirement.
|
3年前 |
Glenn Jocher
|
e5b0200cd2
|
Update tensorboard>=2.4.1 (#2576)
* Update tensorboard>=2.4.1
Update tensorboard version to attempt to address https://github.com/ultralytics/yolov5/issues/2573 (tensorboard logging fail in Docker image).
* cleanup
|
3年前 |
Glenn Jocher
|
ace3e02e40
|
Update requirements.txt (#2173)
|
3年前 |
Glenn Jocher
|
3551b072b3
|
Docker pyYAML>=5.3.1 fix (#2031)
|
3年前 |
Glenn Jocher
|
c76c607265
|
PyYAML==5.4.1 (#2030)
|
3年前 |
Glenn Jocher
|
9a3da79b4a
|
Update requirements.txt (#2021)
* Update requirements.txt
* Update ci-testing.yml
* Update hubconf.py
|
3年前 |
Durgesh
|
ab0db8d104
|
Fix: Seaborn error due to histplot (#1740)
|
3年前 |
Glenn Jocher
|
fa8f1fb0e9
|
Simplify autoshape() post-process (#1653)
* Simplify autoshape() post-process
* cleanup
* cleanup
|
3年前 |
Glenn Jocher
|
0a3ff71ae0
|
Confusion matrix (#1474)
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* Cat apriori to autolabels
* cleanup
|
4年前 |
Glenn Jocher
|
93fc607f07
|
PyTorch Hub and autoShape update (#1415)
|
4年前 |
Glenn Jocher
|
177f39e98e
|
onnx>=1.8.0
|
4年前 |
Glenn Jocher
|
194f16844e
|
Multi-threaded image caching
|
4年前 |
Ayush Chaurasia
|
ca290dca24
|
Weights & Biases (W&B) Feature Addition (#1235)
* Add wandb metric logging and bounding box debugging
* Improve formatting, readability
* Remove mutliple path for init, improve formatting
* Add wandb params
* Remove typecasting in bbox coordinates and reformat
* Cleanup
* add wandb to requirements.txt
* minor updates to test.py
* general reorg
* reduce --log-imgs to 10
* clean wandb import
* reverse wandb import assert
* add except AssertionError to try import
* move wandb init to all global ranks
* replace print() with logger.info()
* replace print() with logger.info()
* move wandb.init() bug fix
* project PosixPath to basename bug fix
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
4年前 |
Glenn Jocher
|
d87cf7ecb9
|
Update requirements.txt coremltools==4.0
|
4年前 |
Glenn Jocher
|
35fe98543c
|
Update requirements.txt coremltools==4.0b4
|
4年前 |
Glenn Jocher
|
987c226849
|
seaborn plotting correlogram addition
|
4年前 |
Glenn Jocher
|
5f07782f56
|
Update requirements.txt for coremltools kmeans quantization
|
4年前 |
Glenn Jocher
|
fb05681a04
|
Update requirements.txt to coremltools==4.0b3
|
4年前 |
Glenn Jocher
|
7b2b52193d
|
update requirements.txt
|
4年前 |
Glenn Jocher
|
e742dcf417
|
update requirements.txt
|
4年前 |
Glenn Jocher
|
61b5733c36
|
update requirements.txt
|
4年前 |
Glenn Jocher
|
3e5e44499d
|
update requirements.txt
|
4年前 |
Glenn Jocher
|
60d5dc0294
|
PyTorch 1.6.0 update with native AMP (#573)
|
4年前 |
Glenn Jocher
|
07a82f4d44
|
Update ci-testing.yml (#445)
* Update ci-testing.yml
* Update ci-testing.yml
* Update requirements.txt
* Update requirements.txt
* Update google_utils.py
* Update test.py
* Update ci-testing.yml
|
4年前 |
Glenn Jocher
|
b6fe2e4595
|
Update requirements.txt
|
4年前 |
Glenn Jocher
|
3e8c89d5ab
|
update requirements.txt
|
4年前 |
Jirka
|
a0fa5f5efe
|
requirements
|
4年前 |
Doug
|
7d19f98273
|
Update requirements.txt
numpy 1.17.3 is required for python3.8
|
4年前 |
Glenn Jocher
|
1b9e28e72e
|
Update requirements.txt
|
4年前 |
Glenn Jocher
|
24dd150fbd
|
update torch>=1.4
|
4年前 |
Lornatang
|
899f1d4bde
|
Fix DDP bug in single process multiple device use cases
|
4年前 |
Prashant Rai
|
0071b69994
|
onnx should be 1.6.0
for smoothly export model to onnx, onnx should be 1.6.0
|
4年前 |
Glenn Jocher
|
260b1729f0
|
FP16 inference update
|
4年前 |
Glenn Jocher
|
629d3701fc
|
pyyaml >= 5.3
|
4年前 |
Glenn Jocher
|
4dd75b3635
|
pycocotools git install fix
|
4年前 |
Glenn Jocher
|
1374581929
|
Update requirements.txt
|
4年前 |
Adam
|
faa6aeaa20
|
Fix pip requirements issues
|
4年前 |
Glenn Jocher
|
1e84a23f38
|
initial commit
|
4年前 |