Glenn Jocher
b3e2f4e08d
Eliminate `total_batch_size` variable ( #3697 )
* Eliminate `total_batch_size` variable
* cleanup
* Update train.py
преди 3 години
Glenn Jocher
fad27c0046
Update DDP for `torch.distributed.run` with `gloo` backend ( #3680 )
* Update DDP for `torch.distributed.run`
* Add LOCAL_RANK
* remove opt.local_rank
* backend="gloo|nccl"
* print
* print
* debug
* debug
* os.getenv
* gloo
* gloo
* gloo
* cleanup
* fix getenv
* cleanup
* cleanup destroy
* try nccl
* return opt
* add --local_rank
* add timeout
* add init_method
* gloo
* move destroy
* move destroy
* move print(opt) under if RANK
* destroy only RANK 0
* move destroy inside train()
* restore destroy outside train()
* update print(opt)
* cleanup
* nccl
* gloo with 60 second timeout
* update namespace printing
преди 3 години
Glenn Jocher
5bab9a28e4
Reformat ( #3694 )
преди 3 години
lb-desupervised
bfb2276b1d
Slightly modify CLI execution ( #3687 )
* Slightly modify CLI execution
This simple change makes it easier to run the primary functions of this
repo (train/detect/test) from within Python. An object which represents
`opt` can be constructed and fed to the `main` function of each of these
modules, rather than having to call the lower level functions directly,
or run the module as a script.
* Update export.py
Add CLI parsing update for more convenient module usage within Python.
Co-authored-by: Lewis Belcher <lb@desupervised.io>
преди 3 години
Mai Thanh Minh
bf209f6fe9
Skip HSV augmentation when hyperparameters are [0, 0, 0] ( #3686 )
* Create shortcircuit in augment_hsv when hyperparameter are zero
* implement faster opt-in
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
Glenn Jocher
814806c61d
Update cache check ( #3691 )
Swapped order of operations for faster first per f527704cd3 (r52362419)
преди 3 години
Glenn Jocher
463628a4d8
Update README.md ( #3684 )
преди 3 години
ZouJiu1
2729761458
EMA changes for pre-model's batch_size ( #3681 )
* EMA changes for pre-model's batch_size
* Update train.py
* Update torch_utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
Glenn Jocher
f527704cd3
Cache v0.3: improved corrupt image/label reporting ( #3676 )
* Cache v0.3: improved corrupt image/label reporting
Fix for https://github.com/ultralytics/yolov5/issues/3656#issuecomment-863660899
* cleanup
преди 3 години
Glenn Jocher
2296f1546f
Update `WORLD_SIZE` and `RANK` retrieval ( #3670 )
преди 3 години
Glenn Jocher
045d5d8629
Update TensorBoard ( #3669 )
преди 3 години
Glenn Jocher
fa201f968e
Update `train(hyp, *args)` to accept `hyp` file or dict ( #3668 )
преди 3 години
Glenn Jocher
6d6e2ca65f
Update train.py ( #3667 )
преди 3 години
Glenn Jocher
ac34834563
Update download_weights.sh comment ( #3662 )
преди 3 години
Glenn Jocher
2754adad46
Remove `/weights` directory ( #3659 )
* Remove `/weights` directory
* cleanup
преди 3 години
Glenn Jocher
9b6dba6207
Update `dataset_stats()` to list of dicts ( #3657 )
* Update `dataset_stats()` to list of dicts
@KalenMike
* Update datasets.py
преди 3 години
SpongeBab
df7706d802
Update README.md ( #3650 )
Be more user-friendly to new users
преди 3 години
Mai Thanh Minh
fa29125f18
`is_coco` list fix ( #3646 )
преди 3 години
Glenn Jocher
6187edcb53
Update CONTRIBUTING.md ( #3647 )
преди 3 години
Glenn Jocher
3ce0db89b0
Update CONTRIBUTING.md ( #3645 )
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
преди 3 години
xiaowk5516
d808855f77
Assert non-premature end of JPEG images ( #3638 )
* premature end of JPEG images
* PEP8 reformat
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
Glenn Jocher
bb79e13d52
RUN pip install --no-cache -U torch torchvision ( #3637 )
преди 3 години
Glenn Jocher
6c0e1d9fd7
Update `verify_image_label()` ( #3635 )
преди 3 години
Glenn Jocher
65f81bfefa
Add `**/*.torchscript.pt` ( #3634 )
преди 3 години
Glenn Jocher
de56813ba8
FROM nvcr.io/nvidia/pytorch:21.05-py3 ( #3633 )
преди 3 години
Glenn Jocher
30e4c4f092
Update README.md ( #3624 )
преди 3 години
Glenn Jocher
7d3686a686
Update `check_file()` ( #3622 )
* Update `check_file()`
* Update datasets.py
преди 3 години
Wei Quan
4c5d9bff80
Fix incorrect end epoch comment ( #3612 )
преди 3 години
Ayush Chaurasia
daab682b06
Suppress wandb images size mismatch warning ( #3611 )
* supress wandb images size mismatch warning
* supress wandb images size mismatch warning
* PEP8 reformat and optimize imports
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
Glenn Jocher
239a11c197
Download COCO and VOC by default ( #3608 )
преди 3 години
masood azhar
6062319ec3
Update datasets.py ( #3591 )
* 'changes-in_dataset'
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
Glenn Jocher
f8adee156d
merge develop
преди 3 години
Kalen Michael
3133607560
Simplify README.md ( #3530 )
* Update README.md
* added hosted images
* added new logo
* testing image hosting
* changed svgs to pngs
* removed old header
* Update README.md
* correct colab image source
* splash.jpg
* rocket and W&B fix
* added contributing template
* added social media to top section
* increased size of top social media
* cleanup and updates
* rearrange quickstarts
* API cleanup
* PyTorch Hub cleanup
* Add tutorials
* cleanup
* update CONTRIBUTING.md
* Update README.md
* update wandb link
* Update README.md
* remove tutorials header
* update environments and integrations
* Comment API image
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* double spaces after section
* Update README.md
* Update README.md
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
Glenn Jocher
88b1945241
Delete __init__.py ( #3596 )
преди 3 години
Glenn Jocher
7a565f130a
Update `dataset_stats()` ( #3593 )
@KalenMike this is a PR to add image filenames and labels to our stats dictionary and to save the dictionary to JSON. Save location is next to the train labels.cache file. The single JSON contains all stats for entire dataset.
Usage example:
```python
from utils.datasets import *
dataset_stats('coco128.yaml', verbose=True)
```
преди 3 години
Ayush Chaurasia
53d4fc2e26
W&B: Allow changed in config variable #3588
преди 3 години
Glenn Jocher
4984cf54be
train.py GPU memory fix ( #3590 )
* train.py GPU memory fix
* ema
* cuda
* cuda
* zeros input
* to device
* batch index 0
преди 3 години
Glenn Jocher
e8c5237403
ConfusionMatrix `normalize=True` fix ( #3587 )
преди 3 години
Glenn Jocher
ec2da4a82c
Add ConfusionMatrix `normalize=True` flag ( #3586 )
преди 3 години
Glenn Jocher
46e1fdfbc6
Update stale.yml ( #3585 )
преди 3 години
Glenn Jocher
5c32bd3080
Created using Colaboratory
преди 3 години
Glenn Jocher
53ed872c28
Update export.py, yolo.py `sys.path.append()` ( #3579 )
преди 3 години
Glenn Jocher
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
* Ignore Seaborn plot warnings
* Update plots.py
* Update metrics.py
преди 3 години
Glenn Jocher
4695ca8314
Refactoring cleanup ( #3565 )
* Refactoring cleanup
* Update test.py
преди 3 години
Glenn Jocher
0e5cfdbea7
Refactor models/export.py arguments ( #3564 )
* Refactor models/export.py arguments
* cleanup
* cleanup
преди 3 години
Glenn Jocher
66cf5c28c1
Refactor detect.py arguments ( #3559 )
* Refactor detect.py arguments
@SkalskiP @KalenMike
* unused ok
* comment arguments
преди 3 години
Glenn Jocher
a9553c04a7
Refactor test.py arguments ( #3558 )
* remove opt from test()
* pass kwargs
* update comments
* revert accidental default change
* multiple --img options
* add comments
преди 3 години
dependabot[bot]
c6deb73a89
Bump pip from 18.1 to 19.2 in /utils/google_app_engine ( #3561 )
Bumps [pip](https://github.com/pypa/pip ) from 18.1 to 19.2.
- [Release notes](https://github.com/pypa/pip/releases )
- [Changelog](https://github.com/pypa/pip/blob/main/NEWS.rst )
- [Commits](https://github.com/pypa/pip/compare/18.1...19.2 )
---
updated-dependencies:
- dependency-name: pip
dependency-type: direct:production
...
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
преди 3 години
Glenn Jocher
4ef9261870
Remove redundant speed/study `half` argument ( #3557 )
преди 3 години
Glenn Jocher
5948f20a3d
Update test.py profiling ( #3555 )
* Update test.py profiling
* half_precision to half
* inplace
преди 3 години