f2d97ebb25
Remove DDP MultiHeadAttention fix ( #3768 )
2021-06-25 12:52:05 +02:00
374957317a
Add `xyxy2xywhn()` ( #3765 )
...
* Edit Comments for numpy2torch tensor process
Edit Comments for numpy2torch tensor process
* add xyxy2xywhn
add xyxy2xywhn
* add xyxy2xywhn
* formatting
* pass arguments
pass arguments
* edit comment for xyxy2xywhn()
edit comment for xyxy2xywhn()
* cleanup datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-25 11:47:46 +02:00
03281f8c76
COCO annotations JSON fix ( #3764 )
2021-06-25 01:55:53 +02:00
f79d7479da
Add optional dataset.yaml `path` attribute ( #3753 )
...
* Add optional dataset.yaml `path` attribute
@KalenMike
* pass locals to python scripts
* handle lists
* update coco128.yaml
* Capitalize first letter
* add test key
* finalize GlobalWheat2020.yaml
* finalize objects365.yaml
* finalize SKU-110K.yaml
* finalize SKU-110K.yaml
* finalize VisDrone.yaml
* NoneType fix
* update download comment
* voc to VOC
* update
* update VOC.yaml
* update VOC.yaml
* remove dashes
* delete get_voc.sh
* force coco and coco128 to ../datasets
* Capitalize Argoverse_HD.yaml
* Capitalize Objects365.yaml
* update Argoverse_HD.yaml
* coco segments fix
* VOC single-thread
* update Argoverse_HD.yaml
* update data_dict in test handling
* create root
2021-06-25 01:25:03 +02:00
417a2f425c
Edit comment ( #3759 )
...
edit comment
2021-06-24 15:57:27 +02:00
ae4261c774
Force non-zero hyp evolution weights `w` ( #3748 )
...
Fix for https://github.com/ultralytics/yolov5/issues/3741
2021-06-23 12:56:22 +02:00
fdc22398fa
Create `data/hyps` directory ( #3747 )
2021-06-23 12:49:38 +02:00
63060910a6
Update `check_datasets()` for dynamic unzip path ( #3732 )
...
@KalenMike
2021-06-22 16:05:38 +02:00
9ac7d388a9
Backwards compatible cache version checks ( #3730 )
2021-06-22 13:50:47 +02:00
0e2d0d54d7
Fix typo ( #3729 )
2021-06-22 13:33:38 +02:00
b83e1a4adc
Fix `img2label_paths()` order ( #3720 )
...
* Fix `img2label_paths()` order
* fix, 1
2021-06-21 22:50:56 +02:00
1f69d12591
Update 4 main ops for paths and .run() ( #3715 )
...
* Add yolov5/ to path
* rename functions to run()
* cleanup
* rename fix
* CI fix
* cleanup find models/export.py
2021-06-21 17:25:04 +02:00
75c0ff43af
[x]W&B: Don't resume transfer learning runs ( #3604 )
...
* Allow config cahnge
* Allow val change in wandb config
* Don't resume transfer learning runs
* Add entity in log dataset
2021-06-21 14:00:25 +02:00
e8810a53e8
Update DDP backend `if dist.is_nccl_available()` ( #3705 )
2021-06-20 17:15:42 +02:00
fbf41e0913
Add `train.run()` method ( #3700 )
...
* Update train.py explicit arguments
* Update train.py
* Add run method
2021-06-20 15:06:58 +02:00
c1af67dcd4
Add torch DP warning ( #3698 )
2021-06-19 19:50:46 +02:00
b3e2f4e08d
Eliminate `total_batch_size` variable ( #3697 )
...
* Eliminate `total_batch_size` variable
* cleanup
* Update train.py
2021-06-19 19:14:59 +02:00
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
2021-06-19 16:30:25 +02:00
5bab9a28e4
Reformat ( #3694 )
2021-06-19 14:21:28 +02:00
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>
2021-06-19 12:06:59 +02:00
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>
2021-06-19 11:51:21 +02:00
814806c61d
Update cache check ( #3691 )
...
Swapped order of operations for faster first per f527704cd3 (r52362419)
2021-06-19 11:22:09 +02:00
463628a4d8
Update README.md ( #3684 )
2021-06-18 17:12:42 +02:00
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>
2021-06-18 16:26:52 +02:00
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
2021-06-18 10:21:47 +02:00
2296f1546f
Update `WORLD_SIZE` and `RANK` retrieval ( #3670 )
2021-06-17 23:24:30 +02:00
045d5d8629
Update TensorBoard ( #3669 )
2021-06-17 22:12:42 +02:00
fa201f968e
Update `train(hyp, *args)` to accept `hyp` file or dict ( #3668 )
2021-06-17 22:03:25 +02:00
6d6e2ca65f
Update train.py ( #3667 )
2021-06-17 21:32:39 +02:00
ac34834563
Update download_weights.sh comment ( #3662 )
2021-06-17 16:21:10 +02:00
2754adad46
Remove `/weights` directory ( #3659 )
...
* Remove `/weights` directory
* cleanup
2021-06-17 14:44:12 +02:00
9b6dba6207
Update `dataset_stats()` to list of dicts ( #3657 )
...
* Update `dataset_stats()` to list of dicts
@KalenMike
* Update datasets.py
2021-06-17 13:59:52 +02:00
df7706d802
Update README.md ( #3650 )
...
Be more user-friendly to new users
2021-06-17 12:37:53 +02:00
fa29125f18
`is_coco` list fix ( #3646 )
2021-06-16 22:56:16 +02:00
6187edcb53
Update CONTRIBUTING.md ( #3647 )
2021-06-16 19:57:11 +02:00
3ce0db89b0
Update CONTRIBUTING.md ( #3645 )
...
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
2021-06-16 19:48:30 +02:00
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>
2021-06-16 13:31:26 +02:00
bb79e13d52
RUN pip install --no-cache -U torch torchvision ( #3637 )
2021-06-16 12:35:33 +02:00
6c0e1d9fd7
Update `verify_image_label()` ( #3635 )
2021-06-16 11:12:15 +02:00
65f81bfefa
Add `**/*.torchscript.pt` ( #3634 )
2021-06-16 10:49:50 +02:00
de56813ba8
FROM nvcr.io/nvidia/pytorch:21.05-py3 ( #3633 )
2021-06-16 10:40:31 +02:00
30e4c4f092
Update README.md ( #3624 )
2021-06-15 15:44:10 +02:00
7d3686a686
Update `check_file()` ( #3622 )
...
* Update `check_file()`
* Update datasets.py
2021-06-15 13:21:04 +02:00
4c5d9bff80
Fix incorrect end epoch comment ( #3612 )
2021-06-15 11:24:56 +02:00
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>
2021-06-14 18:54:58 +02:00
239a11c197
Download COCO and VOC by default ( #3608 )
2021-06-14 13:58:54 +02:00
6062319ec3
Update datasets.py ( #3591 )
...
* 'changes-in_dataset'
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-14 12:28:18 +02:00
f8adee156d
merge develop
2021-06-14 10:45:15 +02:00
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>
2021-06-13 02:37:20 +02:00
88b1945241
Delete __init__.py ( #3596 )
2021-06-12 15:21:37 +02:00
Glenn Jocher
Yonghye Kwon
fcakyon
Ayush Chaurasia
lb-desupervised
Mai Thanh Minh
ZouJiu1
SpongeBab
xiaowk5516
Wei Quan
masood azhar
Kalen Michael