Glenn Jocher
bd581b330b
Create hyp.scratch-p6.yaml ( #3787 )
2021-06-26 15:13:06 +02:00
Glenn Jocher
8035b61682
Update objectness IoU sort ( #3786 )
2021-06-26 14:52:18 +02:00
Glenn Jocher
157aa2f886
Objectness IoU Sort ( #3610 )
...
Co-authored-by: U-LAPTOP-5N89P8V7\banhu <ban.huang@foxmail.com>
2021-06-26 14:45:53 +02:00
Glenn Jocher
9dc5d35fce
Update README.md fix banner width ( #3785 )
2021-06-26 13:05:46 +02:00
Glenn Jocher
f89941711c
NGA xView 2018 Dataset Auto-Download ( #3775 )
...
* update clip_coords for numpy
* uncomment
* cleanup
* Add autosplits
* fix
* cleanup
2021-06-26 00:49:05 +02:00
Ayush Chaurasia
ffb6e11050
W&B: Update Tables API and comply with new dataset_check ( #3772 )
...
* Update tables API and windows path fix
* update dataset check
2021-06-25 20:45:33 +02:00
Piotr Skalski
09246a5a33
fix/incorrect_fitness_import ( #3770 )
2021-06-25 16:16:18 +02:00
Glenn Jocher
f2d97ebb25
Remove DDP MultiHeadAttention fix ( #3768 )
2021-06-25 12:52:05 +02:00
Yonghye Kwon
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
Glenn Jocher
03281f8c76
COCO annotations JSON fix ( #3764 )
2021-06-25 01:55:53 +02:00
Glenn Jocher
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
Yonghye Kwon
417a2f425c
Edit comment ( #3759 )
...
edit comment
2021-06-24 15:57:27 +02:00
Glenn Jocher
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
Glenn Jocher
fdc22398fa
Create data/hyps directory ( #3747 )
2021-06-23 12:49:38 +02:00
Glenn Jocher
63060910a6
Update check_datasets() for dynamic unzip path ( #3732 )
...
@KalenMike
2021-06-22 16:05:38 +02:00
Glenn Jocher
9ac7d388a9
Backwards compatible cache version checks ( #3730 )
2021-06-22 13:50:47 +02:00
fcakyon
0e2d0d54d7
Fix typo ( #3729 )
2021-06-22 13:33:38 +02:00
Glenn Jocher
b83e1a4adc
Fix img2label_paths() order ( #3720 )
...
* Fix `img2label_paths()` order
* fix, 1
2021-06-21 22:50:56 +02:00
Glenn Jocher
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
Ayush Chaurasia
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
Glenn Jocher
e8810a53e8
Update DDP backend if dist.is_nccl_available() ( #3705 )
2021-06-20 17:15:42 +02:00
Glenn Jocher
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
Glenn Jocher
c1af67dcd4
Add torch DP warning ( #3698 )
2021-06-19 19:50:46 +02:00
Glenn Jocher
b3e2f4e08d
Eliminate total_batch_size variable ( #3697 )
...
* Eliminate `total_batch_size` variable
* cleanup
* Update train.py
2021-06-19 19:14:59 +02:00
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
2021-06-19 16:30:25 +02:00
Glenn Jocher
5bab9a28e4
Reformat ( #3694 )
2021-06-19 14:21:28 +02:00
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>
2021-06-19 12:06:59 +02:00
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>
2021-06-19 11:51:21 +02:00
Glenn Jocher
814806c61d
Update cache check ( #3691 )
...
Swapped order of operations for faster first per f527704cd3 (r52362419)
2021-06-19 11:22:09 +02:00
Glenn Jocher
463628a4d8
Update README.md ( #3684 )
2021-06-18 17:12:42 +02:00
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>
2021-06-18 16:26:52 +02:00
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
2021-06-18 10:21:47 +02:00
Glenn Jocher
2296f1546f
Update WORLD_SIZE and RANK retrieval ( #3670 )
2021-06-17 23:24:30 +02:00
Glenn Jocher
045d5d8629
Update TensorBoard ( #3669 )
2021-06-17 22:12:42 +02:00
Glenn Jocher
fa201f968e
Update train(hyp, *args) to accept hyp file or dict ( #3668 )
2021-06-17 22:03:25 +02:00
Glenn Jocher
6d6e2ca65f
Update train.py ( #3667 )
2021-06-17 21:32:39 +02:00
Glenn Jocher
ac34834563
Update download_weights.sh comment ( #3662 )
2021-06-17 16:21:10 +02:00
Glenn Jocher
2754adad46
Remove /weights directory ( #3659 )
...
* Remove `/weights` directory
* cleanup
2021-06-17 14:44:12 +02:00
Glenn Jocher
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
SpongeBab
df7706d802
Update README.md ( #3650 )
...
Be more user-friendly to new users
2021-06-17 12:37:53 +02:00
Mai Thanh Minh
fa29125f18
is_coco list fix (#3646 )
2021-06-16 22:56:16 +02:00
Glenn Jocher
6187edcb53
Update CONTRIBUTING.md ( #3647 )
2021-06-16 19:57:11 +02:00
Glenn Jocher
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
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>
2021-06-16 13:31:26 +02:00
Glenn Jocher
bb79e13d52
RUN pip install --no-cache -U torch torchvision ( #3637 )
2021-06-16 12:35:33 +02:00
Glenn Jocher
6c0e1d9fd7
Update verify_image_label() ( #3635 )
2021-06-16 11:12:15 +02:00
Glenn Jocher
65f81bfefa
Add **/*.torchscript.pt ( #3634 )
2021-06-16 10:49:50 +02:00
Glenn Jocher
de56813ba8
FROM nvcr.io/nvidia/pytorch:21.05-py3 ( #3633 )
2021-06-16 10:40:31 +02:00
Glenn Jocher
30e4c4f092
Update README.md ( #3624 )
2021-06-15 15:44:10 +02:00
Glenn Jocher
7d3686a686
Update check_file() ( #3622 )
...
* Update `check_file()`
* Update datasets.py
2021-06-15 13:21:04 +02:00
