2c073cd207
Add train.py ``--img-size` floor ( #4099 )
2021-07-21 16:50:47 +02:00
0cc7c58787
W&B: fix refactor bugs ( #4069 )
2021-07-19 13:57:13 +02:00
7fdcc77bf4
Albumentations >= 1.0.3 ( #4068 )
2021-07-19 13:23:19 +02:00
c8a98cb7cb
Missing `nc` and `names` handling in check_dataset() ( #4066 )
2021-07-19 13:10:21 +02:00
f7d8562060
`val.py` refactor ( #4053 )
...
* val.py refactor
* cleanup
* cleanup
* cleanup
* cleanup
* save after eval
* opt.imgsz bug fix
* wandb refactor
* dataloader to train_loader
* capitalize global variables
* runs/hub/exp to runs/detect/exp
* refactor wandb logging
* Refactor wandb operations (#4061 )
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2021-07-19 10:43:01 +02:00
62409eea08
W&B sweeps support ( #3938 )
...
* Add support for W&B Sweeps
* Update and reformat
* Update search space
* reformat
* reformat sweep.py
* Update sweep.py
* Move sweeps files to wandb dir
* Remove print
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-14 19:13:02 +02:00
720aaa65c8
Rename `test.py` to `val.py` ( #4000 )
2021-07-14 15:43:54 +02:00
b3dabdcc38
Update `probability` to `p` ( #3980 )
2021-07-12 15:54:43 +02:00
90e60b403d
Update `np.random.random()` to `random.random()` ( #3967 )
2021-07-11 16:56:36 +02:00
8298ce5e88
Update `albumentations>=1.0.2` ( #3966 )
2021-07-11 15:49:30 +02:00
80299a57e2
Numerical stability fix for Albumentations ( #3958 )
2021-07-10 19:50:53 +02:00
443af8b25a
Cache v0.4 update ( #3954 )
2021-07-10 14:18:46 +02:00
a26e7de2bf
Update augmentations.py ( #3948 )
2021-07-09 16:45:04 +02:00
248504cf13
Feature visualization improvements 32 ( #3947 )
2021-07-09 15:23:02 +02:00
8c6f9e15bf
Update `dataset_stats()` for zipped datasets ( #3926 )
...
* Update `dataset_stats()` for zipped datasets
@KalenMike
* cleanup
2021-07-08 11:42:30 +02:00
850970e081
`feature_visualization()` CUDA fix ( #3925 )
2021-07-07 16:23:31 +02:00
87b094bcbc
Feature visualization update ( #3920 )
...
* Feature visualization update
* Save to jpg (faster)
* Save to png
2021-07-07 15:41:58 +02:00
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
2021-07-05 18:01:54 +02:00
6a3ee7cf03
Hub models `map_location=device` ( #3894 )
...
* Hub models `map_location=device`
* cleanup
2021-07-05 16:20:46 +02:00
3c3f8fbd5d
Improved BGR2RGB speeds ( #3880 )
...
* Update BGR2RGB ops
* speed improvements
* cleanup
2021-07-04 20:12:32 +02:00
9e8fb9fd0b
Create `utils/augmentations.py` ( #3877 )
...
* Create `utils/augmentations.py`
* cleanup
2021-07-04 18:14:04 +02:00
9d86b54eb3
Add multi-stream saving feature ( #3864 )
...
* Added the recording feature for multiple streams
Thanks for the very cool repo!!
I was trying to record multiple feeds at the same time, but the current version of the detector only had one video writer and one vid_path!
So the streams were not being saved and only were initialized with one frame and this process didn't record the whole thing.
Fix:
I made a list of `vid_writer` and `vid_path` and the `i` from the loop over the `pred` took care of the writer which need to work!
I hope this helps, Thanks!
* Cleanup list lengths
* batch size variable
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-04 12:55:57 +02:00
831773f5a2
Add EXIF rotation to YOLOv5 Hub inference ( #3852 )
...
* rotating an image according to its exif tag
* Update common.py
* Update datasets.py
* Update datasets.py
faster
* delete extraneous gpg file
* Update common.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-02 13:25:54 +02:00
c6c88dc601
Copy-Paste augmentation for YOLOv5 ( #3845 )
...
* Copy-paste augmentation initial commit
* if any segments
* Add obscuration rejection
* Add copy_paste hyperparameter
* Update comments
2021-07-01 00:35:04 +02:00
7d6af69638
Fix `LoadStreams()` dataloader frame skip issue ( #3833 )
...
* Update datasets.py to read every 4th frame of streams
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-30 12:11:29 +02:00
5ea771d93d
Move IoU functions to metrics.py ( #3820 )
2021-06-29 13:18:13 +02:00
3213d8713f
Fix for `dataset_stats()` with updated data.yaml ( #3819 )
...
@KalenMike
2021-06-29 12:44:59 +02:00
02719dde52
Update `feature_visualization()` ( #3807 )
...
* Update `feature_visualization()`
Only plot for data with height, width > 1
* cleanup
* Cleanup
2021-06-28 13:48:14 +02:00
20d45aa4f1
Add feature map visualization ( #3804 )
...
* Add feature map visualization
Add a feature_visualization function to visualize the mid feature map of the model.
* Update yolo.py
* remove boolean from forward and reorder if statement
* remove print from forward
* General cleanup
* Indent
* Update plots.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-28 13:18:45 +02:00
8e7f285051
Fix datasets for aws and get_coco.sh ( #3788 )
...
* merge master
* Update get_coco.sh
2021-06-26 15:29:16 +02:00
8035b61682
Update objectness IoU sort ( #3786 )
2021-06-26 14:52:18 +02:00
157aa2f886
Objectness IoU Sort ( #3610 )
...
Co-authored-by: U-LAPTOP-5N89P8V7\banhu <ban.huang@foxmail.com>
2021-06-26 14:45:53 +02:00
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
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
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
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
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
b83e1a4adc
Fix `img2label_paths()` order ( #3720 )
...
* Fix `img2label_paths()` order
* fix, 1
2021-06-21 22:50:56 +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
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
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
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
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
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
6c0e1d9fd7
Update `verify_image_label()` ( #3635 )
2021-06-16 11:12:15 +02:00
7d3686a686
Update `check_file()` ( #3622 )
...
* Update `check_file()`
* Update datasets.py
2021-06-15 13:21:04 +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
f8adee156d
merge develop
2021-06-14 10:45:15 +02:00
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)
```
2021-06-12 13:26:41 +02:00
53d4fc2e26
W&B: Allow changed in config variable #3588
2021-06-11 23:18:57 +02:00
e8c5237403
ConfusionMatrix `normalize=True` fix ( #3587 )
2021-06-11 11:46:05 +02:00
ec2da4a82c
Add ConfusionMatrix `normalize=True` flag ( #3586 )
2021-06-11 11:37:08 +02:00
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
...
* Ignore Seaborn plot warnings
* Update plots.py
* Update metrics.py
2021-06-10 13:51:29 +02:00
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>
2021-06-09 21:14:09 +02:00
5948f20a3d
Update test.py profiling ( #3555 )
...
* Update test.py profiling
* half_precision to half
* inplace
2021-06-09 16:25:17 +02:00
8b5086c21b
Multi-GPU default to single device 0 ( #3554 )
...
* Multi-GPU default to single device 0
* Multi-GPU default to single device 0
* add space
2021-06-09 15:31:47 +02:00
958ab92dc1
Remove `opt` from `create_dataloader()`` ( #3552 )
2021-06-09 13:14:56 +02:00
f8ec71e1c2
Update `check_python(minimum=3.6.2)` ( #3548 )
2021-06-09 11:34:15 +02:00
1b5edb6f8e
Update `dataset_stats()` for HUB ( #3536 )
...
* Update `dataset_stats()` for HUB
Cleanup of b6fdd2e
2021-06-09 10:56:11 +02:00
b6fdd2e5e5
Create `dataset_stats()` for HUB
2021-06-08 23:09:45 +02:00
8d52c1c5c5
Update datasets.py ( #3531 )
...
Minor updates to https://github.com/ultralytics/yolov5/pull/3505 , inplace accumulation.
2021-06-08 18:36:40 +02:00
28bff22df8
Use multi-threading in cache_labels ( #3505 )
...
* Use multi threading in cache_labels
* PEP8 reformat
* Add num_threads
* changed ThreadPool.imap_unordered to Pool.imap_unordered
* Remove inplace additions
* Update datasets.py
refactor initial desc
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-08 18:00:21 +02:00
317f2ccc9d
Improved `check_requirements()` offline-handling ( #3466 )
...
Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
2021-06-04 21:34:36 +02:00
563ea9475a
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
2021-06-04 21:17:36 +02:00
c37f072ba7
Faster HSV augmentation ( #3462 )
...
remove datatype conversion process that can be skipped
2021-06-04 20:02:20 +02:00
d40481acc5
Add `timeout()` class ( #3460 )
...
* Add `timeout()` class
* rearrange order
2021-06-04 15:23:33 +02:00
8e3b4a0bf3
Update MixUp augmentation `alpha=beta=32.0` ( #3455 )
...
Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
2021-06-04 12:47:53 +02:00
f8651c388f
Parse URL authentication ( #3424 )
...
* Parse URL authentication
* urllib.parse.unquote()
* improved error handling
* improved error handling
* remove %3F
* update check_file()
2021-06-03 18:44:58 +02:00
3cb9ad4fc4
Update FLOPs description ( #3422 )
...
* Update README.md
* Changing FLOPS to FLOPs.
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
2021-06-03 12:31:51 +02:00
fdbe527dc0
Revert "`cv2.imread(img, -1)` for IMREAD_UNCHANGED ( #3379 )" ( #3395 )
...
This reverts commit 21a9607e00
2021-05-31 10:39:00 +02:00
d833ab3d25
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
```
2021-05-30 20:52:42 +02:00
21a9607e00
`cv2.imread(img, -1)` for IMREAD_UNCHANGED ( #3379 )
...
* Update datasets.py
* comment
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-29 21:12:01 +02:00
ba6f3f974b
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
2021-05-28 15:18:44 +02:00
4d4a2b0520
Ignore blank lines in `*.txt` labels ( #3366 )
...
Fix for https://github.com/ultralytics/yolov5/issues/958#issuecomment-849512083
2021-05-27 14:31:26 +02:00
2435bfe896
Add URL download to check_file() ( #3330 )
...
* Add URL file download to check_file()
* cleanup
* pathlib bug fix
2021-05-26 15:51:49 +02:00
c6b5bfca85
Updated cache v0.2 with `hashlib` ( #3350 )
...
* Update cache v0.2 to include parent hash
Possible fix for https://github.com/ultralytics/yolov5/issues/3349
* Update datasets.py
2021-05-26 14:26:52 +02:00
aad99b63d6
TensorBoard DP/DDP graph fix ( #3325 )
2021-05-25 11:45:24 +02:00
73a92dc1b6
Explicit `git clone` master ( #3311 )
2021-05-24 12:42:36 +02:00
9f3a388cea
Improved check_requirements() robustness ( #3298 )
...
Add try: except clause on missing requirements install to catch install failures. Prompted by Ultralytics YOLOv5 API failure on Cython.
2021-05-23 18:00:23 +02:00
a039b7796d
PyTorch Hub `crops = results.crop()` return values ( #3282 )
...
* Changing save_one_box
Made to work with other changes to common.py
* PEP8 and single line BGR
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-23 16:00:55 +02:00
0e2f2cbb51
Update LoadStreams init fallbacks ( #3295 )
2021-05-23 14:55:42 +02:00
19100ba007
Improves docs and handling of entities and resuming by WandbLogger ( #3264 )
...
* adds latest tag to match wandb defaults
* adds entity handling, 'last' tag
* fixes bug causing finished runs to resume
* removes redundant "last" tag for wandb artifact
2021-05-21 23:42:53 +02:00
683cefead4
YouTube stream ending fix ( #3277 )
...
* YouTube stream ending fix
Properly terminates YouTube streams on video end. Should resolve issues #2769 and #3220 .
* Update datasets.py
2021-05-21 16:51:07 +02:00
f3402353fb
check `batch_size % utilized_device_count` ( #3276 )
...
Bug fix to check batch_size divisibility of utilized CUDA device count vs total system CUDA device count.
2021-05-21 15:20:03 +02:00
7b0eb952ca
`plot_one_box()` default `color=(128, 128, 128)` ( #3240 )
...
* Color can be none by default
* `plot_one_box()` default `color=(128, 128, 128)`
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-19 19:47:36 +02:00
3f74cd9ed1
Parameterize max_det + inference default at 1000 ( #3215 )
...
* Added max_det parameters in various places
* 120 character line
* PEP8
* 120 character line
* Update inference default to 1000 instances
* Update inference default to 1000 instances
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-17 22:47:44 +02:00
13a1c72699
Update datasets.py ( #3216 )
2021-05-17 22:24:26 +02:00
ffb47ffbeb
Silent Patches Bug Fix ( #3214 )
2021-05-17 17:47:30 +02:00
36b0a9e7d0
Add CAP_PROP_FRAME_COUNT for YouTube sources ( #3193 )
2021-05-17 12:27:40 +02:00
ae53f50ae7
Plot labels histogram colors ( #3192 )
2021-05-17 11:44:22 +02:00
c1c7eb023f
Update JSON response ( #3139 )
2021-05-12 18:23:50 +02:00
52c05707d6
Update Colors() ( #3046 )
...
* Update Colors()
* update colors
* update colors
2021-05-12 17:41:11 +02:00
abfcf9eb79
Replace print() with logging.info() in trainloader ( #3103 )
...
Might indirectly help #3095 by providing better visibility on source of corruption.
2021-05-10 17:07:16 +02:00
57b0d3a605
Add check_python() ( #3088 )
...
* Add check_python()
Checks python version against minimum version of 3.7.0.
* remove packaging dependency
* refactor import
2021-05-09 15:33:38 +02:00
d2a17289c9
Explicitly convert artifact path to posix_path ( #3067 )
...
* Explicitly convert artifact path to posix_path
* Remove redudant str() casting
2021-05-07 19:53:56 +02:00
b18ca31ff5
use check_file for --data ( #3035 )
2021-05-05 14:28:19 +02:00
8cab907f60
download() ThreadPool update ( #3027 )
...
* download() ThreadPool update
* update train image count
* cid + 1
2021-05-04 23:37:31 +02:00
5189b3addb
Objects365 Update
2021-05-04 02:25:14 +02:00
41cc7caee6
Add NMS threshold checks ( #3020 )
...
* Add NMS threshold checks
* fix
2021-05-03 15:07:20 +02:00
fcd57028d2
Add is_colab() function ( #3018 )
2021-05-03 12:37:50 +02:00
31ee54c28c
Hubconf.py bug fix ( #3007 )
2021-05-01 18:15:17 +02:00
37eaffec7d
Curl update ( #3004 )
...
* Curl update
* Curl update
2021-05-01 17:29:51 +02:00
4d7bca7648
Add unzip flag to download() ( #3002 )
2021-05-01 12:37:41 +02:00
54652fe3ff
Objects365 update
2021-05-01 00:16:58 +02:00
801b469878
Improve performance of dataset Logger ( #2943 )
...
* Improve performance of Dataset Logger
* Fix scaling bug
2021-04-30 21:54:26 +02:00
955eea8b96
Make cache saving optional ( #2977 )
2021-04-30 00:56:44 +02:00
a833ee2a46
Update check_requirements() exclude list ( #2974 )
2021-04-29 21:16:23 +02:00
2c7c075fb1
Update restapi.py ( #2967 )
2021-04-28 17:50:07 +02:00
57812df68c
New Colors() class ( #2963 )
2021-04-28 16:05:14 +02:00
1849916726
Check_requirements() enclosing apostrophe bug fix ( #2929 )
...
This fixes a bug where the '>' symbol in python package requirements was not running correctly with subprocess.check_output() commands.
2021-04-25 23:33:25 +02:00
9c7bb5a52c
ACON Activation batch-size 1 bug patch ( #2901 )
...
* ACON Activation batch-size 1 bug path
This is not a great solution to https://github.com/nmaac/acon/issues/4 but it's all I could think of at the moment.
WARNING: YOLOv5 models with MetaAconC() activations are incapable of running inference at batch-size 1 properly due to a known bug in https://github.com/nmaac/acon/issues/4 with no known solution.
* Update activations.py
* Update activations.py
* Update activations.py
* Update activations.py
2021-04-25 22:54:07 +02:00
c0d3f80544
Add verbose option to pytorch hub models ( #2926 )
...
* Add verbose and update print to logging
* Fix positonal param
* Revert auto formatting changes
* Update hubconf.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-25 22:01:05 +02:00
45632b2704
Update download() for tar.gz files ( #2919 )
...
* Update download() for tar.gz files
* Update general.py
2021-04-24 15:53:15 +02:00
1b1ab4cca2
Add file_size() function ( #2911 )
...
* Add file_size() function
* Update export.py
2021-04-24 01:31:11 +02:00
14d2d2d75f
Update google_utils.py ( #2900 )
2021-04-22 20:27:32 +02:00
264d860f8d
ACON activation function ( #2893 )
...
* ACON Activation Function
## 🚀 Feature
There is a new activation function [ACON (CVPR 2021)](https://arxiv.org/pdf/2009.04759.pdf ) that unifies ReLU and Swish.
ACON is simple but very effective, code is here: https://github.com/nmaac/acon/blob/main/acon.py#L19
The improvements are very significant:
## Alternatives
It also has an enhanced version meta-ACON that uses a small network to learn beta explicitly, which may influence the speed a bit.
## Additional context
[Code](https://github.com/nmaac/acon ) and [paper](https://arxiv.org/pdf/2009.04759.pdf ).
* Update activations.py
2021-04-22 17:26:05 +02:00
0395e39fd0
Uppercase model filenames enabled ( #2890 )
2021-04-22 16:51:21 +02:00
78fd077657
VisDrone2019-DET Dataset Auto-Download ( #2882 )
...
* VisDrone Dataset Auto-Download
* add visdrone.yaml
* cleanup
* add VisDrone2019-DET-test-dev
* cleanup VOC
2021-04-22 12:10:26 +02:00
d48a34dca7
bug fix: switched rows and cols for correct detections in confusion matrix ( #2883 )
2021-04-22 00:49:55 +02:00
5f7d39fede
Cleanup load_image() ( #2871 )
...
* don't resize up in load_image if augmenting
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-21 15:50:28 +02:00
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
...
* Implement yaml.safe_load()
* yaml.safe_dump()
2021-04-21 14:34:45 +02:00
c949fc86d1
Detection cropping+saving feature addition for detect.py and PyTorch Hub ( #2827 )
...
* Update detect.py
* Update detect.py
* Update greetings.yml
* Update cropping
* cleanup
* Update increment_path()
* Update common.py
* Update detect.py
* Update detect.py
* Update detect.py
* Update common.py
* cleanup
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-20 23:51:08 +02:00
c5c647e281
Update increment_path() to handle file paths ( #2867 )
2021-04-20 19:47:07 +02:00
c15e25c40f
PyTorch Hub cv2 .save() .show() bug fix ( #2831 )
...
* PyTorch Hub cv2 .save() .show() bug fix
cv2.rectangle() was failing on non-contiguous np array inputs. This checks for contiguous arrays and applies is necessary:
```python
imgs[i] = im if im.data.contiguous else np.ascontiguousarray(im) # update
```
* Update plots.py
```python
assert im.data.contiguous, 'Image not contiguous. Apply np.ascontiguousarray(im) to plot_on_box() input image.'
```
* Update hubconf.py
Expand CI tests to OpenCV image.
2021-04-18 13:47:40 +02:00
aff03be35a
YouTube Bug Fix ( #2818 )
...
Fix for #2810
```shell
python detect.py --source 0
```
introduced by YouTube Livestream Detection PR #2752
2021-04-16 17:58:28 +02:00
1f3e482bce
ONNX Simplifier ( #2815 )
...
* ONNX Simplifier
Add ONNX Simplifier to ONNX export pipeline in export.py. Will auto-install onnx-simplifier if onnx is installed but onnx-simplifier is not.
* Update general.py
2021-04-16 14:03:27 +02:00
1479737064
Flask REST API Example ( #2732 )
...
* add files
* Update README.md
* Update README.md
* Update restapi.py
pretrained=True and model.eval() are used by default when loading a model now, so no need to call them manually.
* PEP8 reformat
* PEP8 reformat
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-15 13:26:08 +02:00
0f395b3e3b
YOLOv5 v5.0 Release patch 1 ( #2764 )
...
* torch.jit.trace(model, img, strict=False)
* Update check_file()
* Update hubconf.py
* Update README.md
2021-04-11 23:11:43 +02:00
f5b8f7d54c
YOLOv5 v5.0 Release ( #2762 )
2021-04-11 19:23:47 +02:00
e2b7bc0b32
YouTube Livestream Detection ( #2752 )
...
* Youtube livestream detection
* dependancy update to auto install pafy
* Remove print
* include youtube_dl in deps
* PEP8 reformat
* youtube url check fix
* reduce lines
* add comment
* update check_requirements
* stream framerate fix
* Update README.md
* cleanup
* PEP8
* remove cap.retrieve() failure code
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-11 18:53:40 +02:00
0cae7576a9
utils/wandb_logging PEP8 reformat ( #2755 )
...
* wandb_logging PEP8 reformat
* Update wandb_utils.py
2021-04-10 21:09:23 +02:00
c03d590320
Add Hub results.pandas() method ( #2725 )
...
* Add Hub results.pandas() method
New method converts results from torch tensors to pandas DataFrames with column names.
This PR may partially resolve issue https://github.com/ultralytics/yolov5/issues/2703
```python
results = model(imgs)
print(results.pandas().xyxy[0])
xmin ymin xmax ymax confidence class name
0 57.068970 391.770599 241.383545 905.797852 0.868964 0 person
1 667.661255 399.303589 810.000000 881.396667 0.851888 0 person
2 222.878387 414.774231 343.804474 857.825073 0.838376 0 person
3 4.205386 234.447678 803.739136 750.023376 0.658006 5 bus
4 0.000000 550.596008 76.681190 878.669922 0.450596 0 person
```
* Update comments
torch example input now shown resized to size=640 and also now a multiple of P6 stride 64 (see https://github.com/ultralytics/yolov5/issues/2722#issuecomment-814785930 )
* apply decorators
* PEP8
* Update common.py
* pd.options.display.max_columns = 10
* Update common.py
2021-04-07 16:28:07 +02:00
3067429307
Add support for list-of-directory data format for wandb ( #2719 )
2021-04-06 16:57:13 +02:00
514ebcdf33
Fix : #2674 ( #2683 )
...
* Set resume flag to false
* Check existance of val dataset
2021-04-02 11:54:50 +02:00
2a28ef374b
Set resume flag to false ( #2657 )
2021-03-31 13:47:54 +02:00
9c803f2f7e
Add --label-smoothing eps argument to train.py (default 0.0) ( #2344 )
...
* Add label smoothing option
* Correct data type
* add_log
* Remove log
* Add log
* Update loss.py
remove comment (too versbose)
Co-authored-by: phattran <phat.tranhoang@cyberlogitec.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-03-29 18:45:46 +02:00
866bc7d640
Speed profiling improvements ( #2648 )
...
* Speed profiling improvements
* Update torch_utils.py
deepcopy() required to avoid adding elements to model.
* Update torch_utils.py
2021-03-29 15:19:07 +02:00
1e8ab3f5f2
Add tqdm pbar.close() ( #2644 )
...
When using tqdm, sometimes it can't print in one line and roll to next line.
2021-03-29 12:21:25 +02:00
2e95cf3d79
Improve git_describe() fix 1 ( #2635 )
...
Add stderr=subprocess.STDOUT to catch error messages.
2021-03-28 17:09:06 +02:00
518c09578e
W&B resume ddp from run link fix ( #2579 )
...
* W&B resume ddp from run link fix
* Native DDP W&B support for training, resuming
2021-03-28 16:11:36 +02:00
6e8c5b7678
Improve git_describe() ( #2633 )
...
Catch 'fatal: not a git repository' returns and return '' instead (observed in GCP Hub checks).
2021-03-28 15:39:31 +02:00
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
...
```
2021-03-26 12:52:47 +01:00
196bf10603
Add '*.mpo' to supported image formats ( #2615 )
...
Co-authored-by: Max Uppenkamp <max.uppenkamp@inform-software.com>
2021-03-26 12:45:22 +01:00
77415a42e5
Update git_describe() for remote dir usage ( #2606 )
2021-03-26 01:44:00 +01:00
3bb414890a
resume.py typo ( #2603 )
2021-03-25 20:55:20 +01:00
d4456e43b2
Update segment2box() comment ( #2600 )
2021-03-25 15:12:49 +01:00
333ccc5b0f
YOLOv5 PyTorch Hub models >> check_requirements() ( #2592 )
...
Improved user-feedback following requirements auto-update.
2021-03-24 17:51:39 +01:00
75feeb797c
YOLOv5 PyTorch Hub models >> check_requirements() ( #2591 )
...
Prints 'Please restart runtime or rerun command for update to take effect.' following package auto-install to inform users to restart/rerun.
2021-03-24 16:42:54 +01:00
2b329b0945
Enhanced check_requirements() with auto-install ( #2575 )
...
* Update check_requirements() with auto-install
This PR builds on an idea I had to automatically install missing dependencies rather than simply report an error message.
YOLOv5 should now 1) display all dependency issues and not simply display the first missing dependency, and 2) attempt to install/update each missing/VersionConflict package.
* cleanup
* cleanup 2
* Check requirements.txt file exists
* cleanup 3
2021-03-24 01:05:59 +01:00
1bf9365280
W&B DDP fix ( #2574 )
2021-03-23 16:54:34 +01:00
e8fc97aa38
Improved W&B integration ( #2125 )
...
* Init Commit
* new wandb integration
* Update
* Use data_dict in test
* Updates
* Update: scope of log_img
* Update: scope of log_img
* Update
* Update: Fix logging conditions
* Add tqdm bar, support for .txt dataset format
* Improve Result table Logger
* Init Commit
* new wandb integration
* Update
* Use data_dict in test
* Updates
* Update: scope of log_img
* Update: scope of log_img
* Update
* Update: Fix logging conditions
* Add tqdm bar, support for .txt dataset format
* Improve Result table Logger
* Add dataset creation in training script
* Change scope: self.wandb_run
* Add wandb-artifact:// natively
you can now use --resume with wandb run links
* Add suuport for logging dataset while training
* Cleanup
* Fix: Merge conflict
* Fix: CI tests
* Automatically use wandb config
* Fix: Resume
* Fix: CI
* Enhance: Using val_table
* More resume enhancement
* FIX : CI
* Add alias
* Get useful opt config data
* train.py cleanup
* Cleanup train.py
* more cleanup
* Cleanup| CI fix
* Reformat using PEP8
* FIX:CI
* rebase
* remove uneccesary changes
* remove uneccesary changes
* remove uneccesary changes
* remove unecessary chage from test.py
* FIX: resume from local checkpoint
* FIX:resume
* FIX:resume
* Reformat
* Performance improvement
* Fix local resume
* Fix local resume
* FIX:CI
* Fix: CI
* Imporve image logging
* (:(:Redo CI tests:):)
* Remember epochs when resuming
* Remember epochs when resuming
* Update DDP location
Potential fix for #2405
* PEP8 reformat
* 0.25 confidence threshold
* reset train.py plots syntax to previous
* reset epochs completed syntax to previous
* reset space to previous
* remove brackets
* reset comment to previous
* Update: is_coco check, remove unused code
* Remove redundant print statement
* Remove wandb imports
* remove dsviz logger from test.py
* Remove redundant change from test.py
* remove redundant changes from train.py
* reformat and improvements
* Fix typo
* Add tqdm tqdm progress when scanning files, naming improvements
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-03-23 00:44:50 +01:00
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
2021-03-14 23:16:17 -07:00
2d41e70e82
Scipy kmeans-robust autoanchor update ( #2470 )
...
Fix for https://github.com/ultralytics/yolov5/issues/2394
2021-03-14 21:58:12 -07:00
38ff499b26
Update autosplit() with annotated_only option ( #2466 )
...
* Be able to create dataset from annotated images only
Add the ability to create a dataset/splits only with images that have an annotation file, i.e a .txt file, associated to it. As we talked about this, the absence of a txt file could mean two things:
* either the image wasn't yet labelled by someone,
* either there is no object to detect.
When it's easy to create small datasets, when you have to create datasets with thousands of images (and more coming), it's hard to track where you at and you don't want to wait to have all of them annotated before starting to train. Which means some images would lack txt files and annotations, resulting in label inconsistency as you say in #2313 . By adding the annotated_only argument to the function, people could create, if they want to, datasets/splits only with images that were labelled, for sure.
* Cleanup and update print()
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-03-14 17:11:27 -07:00
08d4918d7f
labels.jpg class names ( #2454 )
...
* labels.png class names
* fontsize=10
2021-03-12 22:15:41 -08:00
f01f3223d5
Integer printout ( #2450 )
...
* Integer printout
* test.py 'Labels'
* Update train.py
2021-03-12 13:35:44 -08:00
e8a2b83268
GCP sudo docker userdata.sh ( #2393 )
...
* GCP sudo docker
* cleanup
2021-03-07 20:21:49 -08:00
7c2c95732c
AWS wait && echo "All tasks done." ( #2391 )
2021-03-07 20:18:30 -08:00
ba18528b47
bbox_iou() stability and speed improvements ( #2385 )
2021-03-06 13:07:34 -08:00
cd8ed3521d
image weights compatible faster random index generator v2 for mosaic augmentation ( #2383 )
...
image weights compatible faster random index generator v2 for mosaic augmentation
2021-03-05 22:58:26 -08:00
2c56ad5436
Confusion matrix background axis swap ( #2114 )
2021-03-02 16:09:52 -08:00
fab5085674
EMA bug fix 2 ( #2330 )
...
* EMA bug fix 2
* update
2021-03-01 17:13:34 -08:00
a82dce7faa
Fix labels being missed when image extension appears twice in filename ( #2300 )
2021-02-25 18:05:38 -08:00
ec1d8496ba
Improved model+EMA checkpointing ( #2292 )
...
* Enhanced model+EMA checkpointing
* update
* bug fix
* bug fix 2
* always save optimizer
* ema half
* remove model.float()
* model half
* carry ema/model in fp32
* rm model.float()
* both to float always
* cleanup
* cleanup
2021-02-24 18:26:46 -08:00
0070995bd5
Amazon AWS EC2 startup and re-startup scripts ( #2282 )
2021-02-24 01:43:59 -08:00
d5d275b6e9
Amazon AWS EC2 startup and re-startup scripts ( #2185 )
...
* Amazon AWS EC2 startup and re-startup scripts
* Create resume.py
* cleanup
2021-02-23 23:10:14 -08:00
83dc1b4484
FLOPS min stride 32 ( #2276 )
...
Signed-off-by: xiaowo1996 <429740343@qq.com>
2021-02-23 09:38:56 -08:00
6f5d6fcdaa
Robust objectness loss balancing ( #2256 )
2021-02-20 11:19:01 -08:00
c09964c27c
Update inference default to multi_label=False ( #2252 )
...
* Update inference default to multi_label=False
* bug fix
* Update plots.py
* Update plots.py
2021-02-19 15:39:09 -08:00
47faf95079
reset head
2021-02-19 15:20:41 -08:00
d2e754b67b
Add isdocker() ( #2232 )
...
* Add isdocker()
* Update general.py
* Update general.py
2021-02-16 15:58:07 -08:00
26c2e54c8f
Add check_imshow() ( #2231 )
...
* Add check_imshow()
* Update general.py
* Update general.py
2021-02-16 13:56:47 -08:00
7b833e37bf
LoadStreams() frame loss bug fix ( #2222 )
2021-02-15 11:02:20 -08:00
bdd88e1ed7
YOLOv5 Segmentation Dataloader Updates ( #2188 )
...
* 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
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* GitHub API rate limit fix
* update
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* astuple
* epochs
* update
* update
* ComputeLoss()
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* merge
* merge
* merge
* merge
* update
* update
* update
* update
* commit=tag == tags[-1]
* Update cudnn.benchmark
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* mosaic9
* update
* update
* update
* update
* update
* update
* institute cache versioning
* only display on existing cache
* reverse cache exists booleans
2021-02-11 21:22:45 -08:00
a5d5f9262d
Changed socket port and added timeout ( #2176 )
2021-02-10 13:01:48 -08:00
c9bda112ae
Update utils/datasets.py to support .webp files ( #2174 )
...
Simply added 'webp' as an image format to the img_formats array so that webp image files can be used as training data.
2021-02-10 00:16:38 -08:00
86897e3663
Update train.py test batch_size ( #2148 )
...
* Update train.py
* Update loss.py
2021-02-06 10:29:32 -08:00
e9b3de463a
Unique *.cache filenames fix ( #2134 )
...
* fix #2121
* Update test.py
* Update train.py
* Update autoanchor.py
* Update datasets.py
* Update log_dataset.py
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-02-05 11:12:43 -08:00
4bdc5a397e
LoadImages() pathlib update ( #2140 )
2021-02-05 09:06:23 -08:00
73a0669930
Start setup for improved W&B integration ( #1948 )
...
* Add helper functions for wandb and artifacts
* cleanup
* Reorganize files
* Update wandb_utils.py
* Update log_dataset.py
We can remove this code, as the giou hyp has been deprecated for a while now.
* Reorganize and update dataloader call
* yaml.SafeLoader
* PEP8 reformat
* remove redundant checks
* Add helper functions for wandb and artifacts
* cleanup
* Reorganize files
* Update wandb_utils.py
* Update log_dataset.py
We can remove this code, as the giou hyp has been deprecated for a while now.
* Reorganize and update dataloader call
* yaml.SafeLoader
* PEP8 reformat
* remove redundant checks
* Update util files
* Update wandb_utils.py
* Remove word size
* Change path of labels.zip
* remove unused imports
* remove --rect
* log_dataset.py cleanup
* log_dataset.py cleanup2
* wandb_utils.py cleanup
* remove redundant id_count
* wandb_utils.py cleanup2
* rename cls
* use pathlib for zip
* rename dataloader to dataset
* Change import order
* Remove redundant code
* remove unused import
* remove unused imports
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-02-01 21:38:41 -08:00
9646ca438a
Update plot_study() ( #2112 )
2021-02-01 13:51:11 -08:00
be9edffded
Update test.py --task speed and study ( #2099 )
...
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
2021-01-31 13:58:49 -08:00
a18efc3a73
Add variable-stride inference support ( #2091 )
2021-01-30 13:47:23 -08:00
630ec06af6
Update to colors.TABLEAU_COLORS ( #2069 )
2021-01-28 12:37:21 -08:00
f639e14e4d
Metric-Confidence plots feature addition ( #2057 )
...
* Metric-Confidence plots feature addition
* cleanup
* Metric-Confidence plots feature addition
* cleanup
* Update run-once lines
* cleanup
* save all 4 curves to wandb
2021-01-27 17:10:53 -08:00
f59f80114c
Add 'exclude' tuple to check_requirements() ( #2041 )
2021-01-26 21:17:36 -08:00
59c21c7bcb
W&B log epoch ( #1946 )
...
* W&B log epoch
* capitalize
* W&B log epoch
* capitalize
* Update train.py
New try using https://docs.wandb.ai/library/log#incremental-logging
* Update train.py
* Update test.py
* Update train.py
* Update plots.py
* Update train.py
* Update train.py
* label plot step -1
* update
* update
* update
* update
* update
* update
* Update train.py
* Update train.py
2021-01-26 21:16:01 -08:00
08d3119e09
Add histogram equalization fcn ( #2049 )
2021-01-26 17:30:42 -08:00
8fcbe43423
Check im.format during dataset caching ( #2042 )
...
* Check im.format during dataset caching
* Update datasets.py
2021-01-25 20:55:35 -08:00
3b7feeafdf
Update general.py check_git_status() fix ( #2020 )
2021-01-22 15:59:01 -08:00
aac33f87ad
Update inference multiple-counting ( #2019 )
...
* Update inference multiple-counting
* update github check
2021-01-22 15:39:08 -08:00
77fd83225f
Update google_utils.py ( #2017 )
2021-01-22 15:08:42 -08:00
1445ab2b27
check_git_status() Windows fix ( #2015 )
...
* check_git_status() Windows fix
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
2021-01-22 12:37:12 -08:00
d9212140b3
Add xywhn2xyxy() ( #1983 )
2021-01-19 13:33:52 -08:00
b1cf25dd9a
check_git_status() asserts ( #1977 )
2021-01-18 10:49:28 -08:00
17751b9891
Update autoanchor.py with yaml.SafeLoader ( #1971 )
2021-01-18 10:47:42 -08:00
35400dc7b7
Update plots.py with yaml.SafeLoader ( #1968 )
2021-01-18 10:46:20 -08:00
b26a2f6242
check_git_status() when not exist /workspace ( #1966 )
2021-01-17 11:55:25 -08:00
e8a41e8916
prevent check_git_status() in docker images ( #1951 )
...
* prevent check_git_status() running docker images
* Update general.py
2021-01-16 01:15:38 -08:00
ca9babb8e6
Add ComputeLoss() class ( #1950 )
2021-01-15 13:50:24 -08:00
b75c432ea0
GitHub API rate limit fallback ( #1930 )
2021-01-13 21:09:50 -08:00
051e9e80de
check_git_status() bug fix ( #1925 )
2021-01-13 10:25:53 -08:00
509dd51aca
check_git_status() improvements ( #1916 )
...
* check_online()
* Update general.py
* update check_git_status()
* reverse rev-parse order
* fetch
* improved responsiveness
* comment
* comment
* remove hyp['giou'] compat warning
2021-01-12 21:51:49 -08:00
dd03b20ba5
colorstr() updates ( #1909 )
...
* W&B ImportError message fix
* colorstr() updates
* colorstr() updates
* colorstr() default to 'blue', 'bold'
* train: magenta
* train: blue
2021-01-12 10:33:15 -08:00
1d1c0567a4
PyTorch Hub results.render() ( #1897 )
2021-01-10 20:06:11 -08:00
ffef77124e
Update torch_utils.py ( #1895 )
2021-01-10 12:26:29 -08:00
0f11aaf551
GitHub API rate limit newline ( #1894 )
2021-01-10 11:47:00 -08:00
d06ad3bc3d
GitHub API rate limit fix ( #1894 )
2021-01-10 11:42:45 -08:00
6ab589583c
Add colorstr() ( #1887 )
...
* Add colorful()
* update
* newline fix
* add git description
* --always
* update loss scaling
* update loss scaling 2
* rename to colorstr()
2021-01-09 15:24:18 -08:00
fea9c9b80c
nn.SiLU() citation correction ( #1713 )
2021-01-06 20:58:41 -08:00
e77c77f580
Add check_requirements() ( #1853 )
...
* Add check_requirements()
* add import
* parameterize filename
* add to detect, test
2021-01-06 16:35:40 -08:00
135ec5c5ce
W&B ID reset on training completion ( #1852 )
...
* Update train.py
Fix the bug of always the same W&B ID and continue overwrite with the old logging.
BUG report
https://github.com/ultralytics/yolov5/issues/1851
* Fix the bug of duplicate W&B ID
fix the bug of https://github.com/ultralytics/yolov5/issues/1851
If we had trained on yolov5s.pt, the program will generate a new unique W&B ID.
If we hadn't, the program will keep the old code, we can still use --resume aug.
* Update general.py
* revert train.py changes
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-01-06 16:27:22 -08:00
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
2021-01-04 19:54:09 -08:00
0e341c5660
Create one_cycle() function ( #1836 )
2021-01-04 15:49:08 -08:00
9f5a18bb80
Torch CUDA synchronize update ( #1826 )
...
* torch.cuda.synchronize() update
* torch.cuda.synchronize() update
* torch.cuda.synchronize() update
* newline
2021-01-03 11:23:12 -08:00
c0ffcdf998
Display correct CUDA devices ( #1776 )
...
* Display correct CUDA devices
* cleanup
2020-12-24 13:01:35 -08:00
3004fb5bc1
Automatic m.half() profile on x.half()
2020-12-21 15:20:33 -08:00
0bd9c48609
Update torch_utils.py
...
FLOPS to GFLOPS
2020-12-21 13:29:52 -08:00
394d1c89f3
Input channel yaml['ch'] addition ( #1741 )
2020-12-19 10:54:01 -08:00
685d601308
Increase plot_labels() speed ( #1736 )
2020-12-18 18:05:38 -08:00
49abc722fc
Update profile_idetection() ( #1727 )
2020-12-18 01:19:17 -08:00
d5289b54c4
clean_str() function addition ( #1674 )
...
* clean_str() function addition
* cleanup
* add euro symbol €
* add closing exclamation (spanish)
* cleanup
2020-12-17 17:20:20 -08:00
6bd5e8bca7
nn.SiLU() export support ( #1713 )
2020-12-16 17:55:57 -08:00
1fc9d42a64
NMS --classes 0 bug fix ( #1710 )
2020-12-16 08:58:51 -08:00
8bc0027afc
Update loss criteria constructor ( #1711 )
2020-12-16 08:39:35 -08:00
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>
2020-12-15 20:42:14 -08:00
69ea70cd3b
Add idetection_profile() function to plots.py ( #1700 )
2020-12-15 18:35:47 -08:00
54043a9fa4
Streaming --save-txt bug fix ( #1672 )
...
* Streaming --save-txt bug fix
* cleanup
2020-12-11 15:45:32 -08:00
ada90e3901
Profile() feature addition ( #1673 )
...
* Profile() feature addition
* cleanup
2020-12-11 09:30:39 -08:00
94a7f55c4e
FReLU bias=False bug fix ( #1666 )
2020-12-10 13:06:15 -08:00
2e8e02745b
vast.ai compatability updates ( #1657 )
2020-12-09 19:01:08 -08:00
fa8f1fb0e9
Simplify autoshape() post-process ( #1653 )
...
* Simplify autoshape() post-process
* cleanup
* cleanup
2020-12-09 07:44:06 -08:00
84f9bb5d92
Normalized mosaic plotting bug fix ( #1647 )
2020-12-08 18:44:13 -08:00
0bb43953eb
Reinstate PR curve sentinel values ( #1645 )
2020-12-08 17:40:49 -08:00
68e6ab668b
Hub device mismatch bug fix ( #1619 )
2020-12-06 17:53:38 +01:00
791dadb51c
Pycocotools best.pt after COCO train ( #1616 )
...
* Pycocotools best.pt after COCO train
* cleanup
2020-12-06 14:58:33 +01:00
8918e63476
Increase FLOPS robustness ( #1608 )
2020-12-05 11:41:34 +01:00
f010147578
Update matplotlib.use('Agg') tight ( #1583 )
...
* Update matplotlib tight_layout=True
* udpate
* udpate
* update
* png to ps
* update
* update
2020-12-02 15:53:16 +01:00
784feae30a
Update matplotlib svg backend ( #1580 )
2020-12-02 14:05:12 +01:00
577f298d9b
plot_images() scale bug fix ( #1566 )
...
fix bugs in plot_images
2020-12-01 11:29:59 +01:00
b6ed1104a6
Daemon thread plotting ( #1561 )
...
* Daemon thread plotting
* remove process_batch
* plot after print
2020-11-30 16:44:14 +01:00
cff9263490
f.read().strip() ( #1551 )
2020-11-29 11:59:52 +01:00
9fa7f9f598
f.read().strip()
2020-11-29 11:58:14 +01:00
96a84468b9
Update labels_to_image_weights() ( #1545 )
2020-11-28 12:25:45 +01:00
c9798ae0e1
Update plot_study_txt() ( #1533 )
2020-11-26 22:18:17 +01:00
0f2057ed33
Targets scaling bug fix ( #1529 )
2020-11-26 18:33:28 +01:00
2c3efa430b
Mosaic plots bug fix ( #1526 )
2020-11-26 14:02:22 +01:00
12499f1c01
--image_weights bug fix ( #1524 )
2020-11-26 13:25:51 +01:00
9728e2b8ae
--image_weights bug fix ( #1524 )
2020-11-26 11:49:01 +01:00
e9a0ae6f19
Cache bug fix ( #1513 )
...
* Caching bug fix #1508
* np.zeros((0,5)) x2
2020-11-25 20:33:14 +01:00
b3ceffb513
Add QFocalLoss() ( #1482 )
...
* Update loss.py
implement the quality focal loss which is a more general case of focal loss
more detail in https://arxiv.org/abs/2006.04388
In the obj loss (or the case cls loss with label smooth), the targets is no long barely be 0 or 1 (can be 0.7), in this case, the normal focal loss is not work accurately
quality focal loss in behave the same as focal loss when the target is equal to 0 or 1, and work accurately when targets in (0, 1)
example:
targets:
tensor([[0.6225, 0.0000, 0.0000],
[0.9000, 0.0000, 0.0000],
[1.0000, 0.0000, 0.0000]])
___________________________
pred_prob:
tensor([[0.6225, 0.2689, 0.1192],
[0.7773, 0.5000, 0.2227],
[0.8176, 0.8808, 0.1978]])
____________________________
focal_loss
tensor([[0.0937, 0.0328, 0.0039],
[0.0166, 0.1838, 0.0199],
[0.0039, 1.3186, 0.0145]])
______________
qfocal_loss
tensor([[7.5373e-08, 3.2768e-02, 3.9179e-03],
[4.8601e-03, 1.8380e-01, 1.9857e-02],
[3.9233e-03, 1.3186e+00, 1.4545e-02]])
we can see that targets[0][0] = 0.6255 is almost the same as pred_prob[0][0] = 0.6225,
the targets[1][0] = 0.9 is greater then pred_prob[1][0] = 0.7773 by 0.1227
however, the focal loss[0][0] = 0.0937 larger then focal loss[1][0] = 0.0166 (which against the purpose of focal loss)
for the quality focal loss , it implement the case of targets not equal to 0 or 1
* Update loss.py
2020-11-25 19:32:27 +01:00
2026d4c5eb
Update caching ( #1496 )
2020-11-24 16:25:21 +01:00
bde5d9aaaa
Update caching ( #1496 )
2020-11-24 16:23:00 +01:00
0822cda781
Update caching ( #1496 )
2020-11-24 16:22:02 +01:00
89c7a5b8dc
Update caching ( #1496 )
2020-11-24 16:13:04 +01:00
44f42b1589
changed prints to logging in utils/datasets ( #1315 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-11-24 16:03:19 +01:00
7aeef2dca5
Prevent PR plotting ( #1489 )
2020-11-24 00:46:01 +01:00
354109c54c
Autosplit ( #1488 )
2020-11-23 18:35:25 +01:00
4798e66fdf
Autosplit ( #1488 )
2020-11-23 17:18:21 +01:00
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
2020-11-23 15:27:14 +01:00
95fa65339f
Cat apriori to autolabels ( #1484 )
2020-11-23 13:38:47 +01:00
201bafc7cf
Sync train and test iou_thresh ( #1465 )
...
* Sync train and test iou_thresh
* Sync train and test iou_thresh
* weights names .lower()
* Notebook update
2020-11-21 12:38:35 +01:00
a70e554fc3
Remove redundant downloads mirror ( #1461 )
2020-11-20 13:10:56 +01:00
394131c2aa
Use torchvision.ops.nms ( #1460 )
2020-11-20 11:23:36 +01:00
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>
2020-11-19 12:56:20 +01:00
64bce3c822
Update labels.png with rectangles fix ( #1432 )
2020-11-18 12:27:30 +01:00
1542ccadb2
Update labels.png with rectangles ( #1432 )
2020-11-18 00:17:33 +01:00
4250f84dfb
Update PR curve ( #1428 )
...
* Update PR curve
* legend outside
* list(Path().glob())
2020-11-17 17:29:55 +01:00
8d2d6d2349
Update img2label_paths() ( #1414 )
2020-11-17 13:26:06 +01:00
f5429260ca
PyTorch Hub and autoShape update ( #1415 )
...
* PyTorch Hub and autoShape update
* comment x for imgs
* reduce comment
2020-11-16 23:09:55 +01:00
92c9b72832
Matplotlib AGG backend ( #1413 )
2020-11-16 16:34:07 +01:00
548a98a391
Recursive directories dataset capability ( #1408 )
...
* Recursive directories dataset capability
* x.split('.')[-1]
* f += glob.glob(f"{p}{os.sep}**.*", recursive=True)
* f += glob.glob(str(p / '**.*'), recursive=True)
* remove . from image and vid formats
* .txt to txt
* str(p / '**' / '*.*')
2020-11-16 16:24:57 +01:00
9c91aeae10
W&B logging add hyperparameters ( #1399 )
...
* W&B logging add hyperparameters
* hyp bug fix and image logging updates
* if plots and wandb:
* cleanup
* wandb/ gitignore add
* cleanup 2
* cleanup 3
* move wandb import to top of file
* wandb evolve
* update import
* wandb.run.finish()
* default anchors: 3
2020-11-16 13:35:34 +01:00
9f27902368
Improved FLOPS computation ( #1398 )
2020-11-14 14:48:55 +01:00
d8f5fcfe87
Improved FLOPS computation ( #1398 )
...
* Improved FLOPS computation
* update comment
2020-11-14 14:39:46 +01:00
fe341fa44d
Utils reorganization ( #1392 )
...
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
2020-11-14 11:50:32 +01:00
c4addd7761
Unified '/project/name' results saving ( #1377 )
...
* Project/name update
* Update ci-testing.yml
* address project with path separator failure mode
* Project/name update
* address project with path separator failure mode
* Update ci-testing.yml
* detect.py default --name bug fix
* missing rstrip PR
* train/exp0 to train/exp
2020-11-12 23:37:46 +01:00
67bf9a974e
Replace 'ground truth' with 'labels' ( #1337 )
...
* Replace 'ground truth' with 'labels'
* Using torch... update
2020-11-10 14:15:03 +01:00
19e2482458
PyTorch version to screen and cleanup ( #1325 )
...
* Create flatten_recursive() helper function
* cleanup
* print torch version
2020-11-09 12:24:11 +01:00
4821d076e2
Increment train, test, detect runs/ ( #1322 )
...
* Increment train, test, detect runs/
* Update ci-testing.yml
* inference/images to data/images
* move images
* runs/exp to runs/train/exp
* update 'results saved to %s' str
2020-11-08 19:39:05 +01:00
d3e7778151
imap bug fix
2020-11-07 02:27:31 +01:00
194f16844e
Multi-threaded image caching
2020-11-07 02:18:18 +01:00
ea7e78cb11
Grid indices overflow bug fix ( #1316 )
2020-11-06 19:17:50 +01:00
fed5d9cdfb
Grid indices overflow bug fix ( #1313 )
2020-11-06 13:40:11 +01:00
2062765e06
auto-download v3.1 models ( #1259 )
2020-11-01 20:11:41 +01:00
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
...
* torch 1.7.0 compatibility updates
* add inference verification
2020-10-28 15:03:50 +01:00
453acdec67
Update tensorboard logging
2020-10-26 01:08:33 +01:00
8e9b3d0205
Precision-Recall Curve feature update #1206
2020-10-25 12:59:16 +01:00
ed85038c38
Precision-Recall Curve feature update ( #1206 )
...
* Precision-Recall Curve feature update
* sentinel value update
2020-10-25 12:55:52 +01:00
481d46cffb
Improved corruption handling during scan and cache ( #999 )
2020-10-24 15:09:19 +02:00
d61930e017
Improved corruption handling during scan and cache ( #999 )
2020-10-24 14:50:50 +02:00
9291daa6f7
fix padding for rectangular inference ( #1165 )
...
Co-authored-by: swsong <swsong@stratosphere.mobi>
2020-10-19 12:22:22 +02:00
3b57cb5641
Simplified inference ( #1153 )
2020-10-15 20:10:08 +02:00
c67e72200e
fix compatibility for hyper config ( #1146 )
...
* fix/hyper
* Hyp giou check to train.py
* restore general.py
* train.py overwrite fix
* restore general.py and pep8 update
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-10-15 15:05:58 +02:00
4d3680c81d
Minor import and spelling updates ( #1133 )
2020-10-13 17:24:27 +02:00
00917a6225
update expt name comment and folder parsing for training ( #978 )
...
* comment
* fix parsing
* fix evolve
* folder
* tqdm
* Update train.py
* Update train.py
* reinstate anchors into meta dict
anchor evolution is working correctly now
* reinstate logger
prefer the single line readout for concise logging, which helps simplify notebook and tutorials etc.
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-10-13 14:10:21 +02:00
0ada058f63
Generalized regression criterion renaming ( #1120 )
2020-10-11 17:25:17 +02:00
5fac5ad165
Precision-Recall Curve Feature Addition ( #1107 )
...
* initial commit
* Update general.py
Indent update
* Update general.py
refactor duplicate code
* 200 dpi
2020-10-09 14:50:59 +02:00
124f0e8212
torchvision nms bug fix
2020-10-06 15:09:24 +02:00
883924d9dc
classifier, export, torch seed updates
2020-10-06 14:54:02 +02:00
c5d2331897
Fix increment_dir to use run_xxx for logdir ( #1058 )
...
* Fix increment_dir to use run_xxx for logdir
Rerunning train.py with logdir in the form of `run_xxx` causes index slicing in
```
n = max([int(x[len(dir):x.find('_') if '_' in Path(x).name else None]) for x in d]) + 1 # increment
```
to underflow.
* Replace find with rfind
2020-10-05 15:46:32 +02:00
7220cee1d1
mosaic4 bottom left image fix ( #1021 )
...
fix a bug in load_mosaic
2020-09-23 09:51:33 -07:00
702c4fa53e
adding the configuration to deploy yolov5 in in app engine ( #964 )
...
* add the configuration to deploy yolov5 in a docker container in app engine on google cloud
* remove the duplicated requirements, place the google app engine files in utils
* remove the unused requirements
Co-authored-by: guilhembau <guilhem@6dbytes.com>
2020-09-22 11:48:44 -07:00
89655a84f2
.fuse() gradient introduction bug fix
2020-09-20 11:57:19 -07:00
5a9c5c1d3b
add mosaic and warmup to hyperparameters ( #931 )
2020-09-13 14:08:06 -07:00
f1c63e2784
add mosaic and warmup to hyperparameters ( #931 )
2020-09-13 14:03:54 -07:00
806e75f2b1
self.label_files bug fix ( #958 )
2020-09-13 11:38:23 -07:00
c54e39464c
check_file() assert unique
2020-09-11 14:33:06 -07:00
720645d96a
LoadImagesAndLabels() /images/ to /labels/ fix ( #928 )
2020-09-10 12:29:05 -07:00
d49c52eee3
_RepeatSampler outside of InfiniteDataLoader
2020-09-10 12:27:35 -07:00
9c0efdde9f
glob.glob(recursive=True) search ( #917 )
2020-09-06 22:52:28 -07:00
bc1fd13a7a
gsutil cp hyp evolution bug fix ( #876 )
2020-09-03 20:29:08 -07:00
44cdcc7e0b
hyp['anchors'] evolution update
2020-09-03 12:54:22 -07:00
ffe9eb4238
remove if platform.system() == 'Darwin' checks
2020-09-03 11:41:33 -07:00
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
2020-09-01 22:18:42 -07:00
d3f9bf2bb7
Update datasets.py
2020-09-01 17:02:47 -07:00
1e15aad6f9
Add InfiniteDataLoader class ( #876 )
...
* Add InfiniteDataLoader
Only initializes at first epoch. Saves time.
* Moved class to a better location
2020-08-31 11:01:25 -07:00
987c226849
seaborn plotting correlogram addition
2020-08-29 15:13:58 -07:00
9776e70988
torch.ops.torchvision.nms ( #860 )
...
Don't load the entire torchvision library just for nms when the function is already in the torch library.
2020-08-27 11:13:41 -07:00
31d6c389f2
Update general.py plot_evolution()
2020-08-25 23:08:14 -07:00
455f7b8f76
Update general.py ( #823 )
...
Fixes #822
`init_seeds` from `torch_utils` import is being overwritten by function `init_seeds` in `general.py`
2020-08-25 13:01:17 -07:00
5a7d79fbe6
Update general.py bbox_iou() with eps improvements ( #736 )
...
eps reduced to 1e-9 for additional precaution.
2020-08-25 12:56:39 -07:00
82ed33a052
generalize webcam detection with opt.source.isnumeric() (yolov3 #1462 )
2020-08-25 11:07:38 -07:00
4d7f222f73
Update export.py with v3.0 Hardswish() support
2020-08-24 21:47:49 -07:00
8666bc507e
Create evolve.sh for infinite hyp evolution
2020-08-24 20:27:28 -07:00
71209a6099
exportable Hardswish() implementation
2020-08-22 16:59:31 -07:00
f7e075fe93
gsutil cp yaml when evolving to GCP bucket
2020-08-22 15:47:39 -07:00
9dbaa1bbbe
Update plot_results()
2020-08-22 15:32:05 -07:00
ea5e97fb9b
FP16 NMS ( #814 )
2020-08-21 14:44:25 -07:00
b42e8a531b
optimize imports
2020-08-20 21:17:40 -07:00
5e0b90de8f
CIoU nan bug fix ( #736 )
2020-08-20 10:37:55 -07:00
ebafd1ead5
single command --resume ( #756 )
...
* single command --resume
* else check files, remove TODO
* argparse.Namespace()
* tensorboard lr
* bug fix in get_latest_run()
2020-08-17 16:28:43 -07:00
26c3b11fdc
Update torch_utils.py ( #751 )
2020-08-16 13:58:59 -07:00
f5da528d28
reformat code
2020-08-14 11:53:44 -07:00
66744a0df1
Update utils/google_utils.py ( #733 )
...
To download weights from v3.0 instead of v2.0
2020-08-13 18:44:14 -07:00
916d4aad9a
v3.0 Release ( #725 )
...
* initial commit
* remove yolov3-spp from test.py study
* update study --img range
* update mAP
* cleanup and speed updates
* update README plot
2020-08-13 14:25:05 -07:00
56c2c344ff
Fix list paths ( #721 )
...
* Add list paths on check_dataset
* missing raise statement
* Update general.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-12 18:53:38 -07:00
a925f283a7
max workers for dataloader ( #722 )
2020-08-12 13:57:36 -07:00
d2da523053
Merge remote-tracking branch 'origin/master'
2020-08-11 20:11:58 -07:00
9f482cbcb8
utils.general comment updates/bug fixes
2020-08-11 20:11:44 -07:00
1f92422e20
Github release assets model autodownload ( #711 )
...
* assets autodownload
* Update google_utils.py
* Update google_utils.py
* Update google_utils.py
* Update google_utils.py
* Update google_utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-11 17:17:24 -07:00
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>
2020-08-11 11:18:45 -07:00
542833c997
Filter the None labels ( #705 )
...
* the value should be a list to unpacking
* filter the None labels
2020-08-11 11:15:48 -07:00
9ae868364a
the value should be a list to unpacking ( #688 )
2020-08-10 11:23:48 -07:00
41523e2c91
Dataset autodownload feature addition ( #685 )
...
* initial commit
* move download scripts into data/scripts
* new check_dataset() function in general.py
* move check_dataset() out of with context
* Update general.py
* DDP update
* Update general.py
2020-08-09 20:52:57 -07:00
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
2020-08-09 02:12:44 -07:00
3fcd365ba3
Fix curl download on Windows ( #669 )
2020-08-07 12:13:55 -07:00
fe5b3f8712
Fix curl download on Windows ( #669 )
...
* Update ci-testing.yml
* Fix windows download
* Fix cookie and curl download issue from gdrive
* Revert "Update ci-testing.yml"
This reverts commit 7389d2238d
2020-08-07 12:10:59 -07:00
7eaf225d55
zero-target training bug fix ( #609 )
2020-08-05 13:35:31 -07:00
f346da9f2b
update activations.py
2020-08-03 17:18:31 -07:00
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
2020-08-02 15:47:36 -07:00
ec7a926163
AutoAnchor update to display anchors/target
2020-08-02 14:38:45 -07:00
d989bc9260
remove NBSP
2020-08-02 14:23:05 -07:00
023e37807c
hyperparameter evolution update ( #566 )
2020-08-02 13:18:03 -07:00
64e6d199ba
hyperparameter evolution update ( #566 )
2020-08-02 13:17:26 -07:00
305c6a028a
compute_loss() leaf variable update
2020-08-02 10:30:43 -07:00
c1a2a7a411
hyperparameter evolution bug fix ( #566 )
2020-08-01 23:00:10 -07:00
e32abb5fb9
hyperparameter evolution bug fix ( #566 )
2020-08-01 19:15:48 -07:00
127cbeb3f5
hyperparameter expansion to flips, perspective, mixup
2020-08-01 13:47:54 -07:00
23d2bde01f
leaf variable --single-cls training bug fix ( #593 )
2020-08-01 11:42:41 -07:00
bcd452c482
replace random_affine() with random_perspective()
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-31 15:53:52 -07:00
bb87276d80
update build_targets() ( #589 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-31 14:34:13 -07:00
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
2020-07-31 10:52:45 -07:00
c8892672a7
Funnel ReLU (FReLU) @staticmethod bug fix ( #556 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 17:38:15 -07:00
131782a2aa
Funnel ReLU (FReLU) ( #556 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 17:35:34 -07:00
4b074d9d9d
Funnel ReLU (FReLU) ( #556 )
...
* fix #541 #542
* Update train.py
* Add Frelu
* Update activations.py
PEP8 and format updates for commonality with models.common.Conv()
* Update activations.py
Update case
* Update activations.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 17:25:49 -07:00
7d78b17a69
Fix utils.py nonzero() warning ( #551 )
2020-07-29 12:14:11 -07:00
a209a32019
Fix bug #541 #542 ( #545 )
...
* fix #541 #542
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-28 18:31:01 -07:00
0032af2980
Update utils.py strip_optimizer() ( #509 )
2020-07-24 19:26:56 -07:00
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
...
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
2020-07-24 11:42:23 -07:00
4b5f4806bc
Update datasets.py ( #494 )
2020-07-23 22:49:54 -07:00
9da56b62dd
v2.0 Release ( #491 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-23 15:34:23 -07:00
4ffd9779d3
plotting improvements ( #471 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-21 14:07:11 -07:00
1e95337f3a
Update datasets.py ( #454 )
2020-07-20 10:33:09 -07:00
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
2020-07-19 15:44:11 -07:00
4102fcc9a7
[WIP] Feature/ddp fixed ( #401 )
...
* Squashed commit of the following:
commit d738487089e41c22b3b1cd73aa7c1c40320a6ebf
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 17:33:38 2020 +0700
Adding world_size
Reduce calls to torch.distributed. For use in create_dataloader.
commit e742dd9619d29306c7541821238d3d7cddcdc508
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 15:38:48 2020 +0800
Make SyncBN a choice
commit e90d4004387e6103fecad745f8cbc2edc918e906
Merge: 5bf8beb cd90360
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 14 15:32:10 2020 +0800
Merge pull request #6 from NanoCode012/patch-5
Update train.py
commit cd9036017e7f8bd519a8b62adab0f47ea67f4962
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 13:39:29 2020 +0700
Update train.py
Remove redundant `opt.` prefix.
commit 5bf8bebe8873afb18b762fe1f409aca116fac073
Merge: c9558a9 a1c8406https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit c9558a9b51547febb03d9c1ca42e2ef0fc15bb31
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 13:51:34 2020 +0800
Add device allocation for loss compute
commit 4f08c692fb5e943a89e0ee354ef6c80a50eeb28d
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:16:27 2020 +0800
Revert drop_last
commit 1dabe33a5a223b758cc761fc8741c6224205a34b
Merge: a1ce9b1 4b8450b
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:49 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit a1ce9b1e96b71d7fcb9d3e8143013eb8cebe5e27
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:21 2020 +0800
fix lr warning
commit 4b8450b46db76e5e58cd95df965d4736077cfb0e
Merge: b9a50ae 02c63ef
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Wed Jul 8 21:24:24 2020 +0800
Merge pull request #4 from NanoCode012/patch-4
Add drop_last for multi gpu
commit 02c63ef81cf98b28b10344fe2cce08a03b143941
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Wed Jul 8 10:08:30 2020 +0700
Add drop_last for multi gpu
commit b9a50aed48ab1536f94d49269977e2accd67748f
Merge: ec2dc6c 121d90bhttps://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit ec2dc6cc56de43ddff939e14c450672d0fbf9b3d
Merge: d0326e3 82a6182
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:34:31 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit d0326e398dfeeeac611ccc64198d4fe91b7aa969
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:31:24 2020 +0800
Add SyncBN
commit 82a6182b3ad0689a4432b631b438004e5acb3b74
Merge: 96fa40a 050b2a5
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 7 19:21:01 2020 +0800
Merge pull request #1 from NanoCode012/patch-2
Convert BatchNorm to SyncBatchNorm
commit 050b2a5a79a89c9405854d439a1f70f892139b1c
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:38:14 2020 +0700
Add cleanup for process_group
commit 2aa330139f3cc1237aeb3132245ed7e5d6da1683
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:07:40 2020 +0700
Remove apex.parallel. Use torch.nn.parallel
For future compatibility
commit 77c8e27e603bea9a69e7647587ca8d509dc1990d
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 01:54:39 2020 +0700
Convert BatchNorm to SyncBatchNorm
commit 96fa40a3a925e4ffd815fe329e1b5181ec92adc8
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 21:53:56 2020 +0800
Fix the datset inconsistency problem
commit 16e7c269d062c8d16c4d4ff70cc80fd87935dc95
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 11:34:03 2020 +0800
Add loss multiplication to preserver the single-process performance
commit e83805563065ffd2e38f85abe008fc662cc17909
Merge: 625bb49 3bdea3fhttps://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit 625bb49f4e52d781143fea0af36d14e5be8b040c
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 2 22:45:15 2020 +0800
DDP established
* Squashed commit of the following:
commit 94147314e559a6bdd13cb9de62490d385c27596f
Merge: 65157e2 37acbdchttps://github.com/ultralytics/yolov4 into feature/DDP_fixed
commit 37acbdc0b6b8c2da4a0dhttps://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit cd55b445c4dcd8003ff4b0b46b64adf7c16e5ce7
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Wed Jul 15 16:42:33 2020 +0800
fix the DDP performance deterioration bug.
commit 0f3b8bb1fae5885474ba861bbbd1924fb622ee93
Author: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed Jul 15 00:28:53 2020 -0700
Delete README.md
commit f5921ba1e35475f24b062456a890238cb7a3cf94
Merge: 85ab2f3 bd3fdbb
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Wed Jul 15 11:20:17 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit bd3fdbbf1b08ef87931eef49fa8340621caa7e87
Author: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue Jul 14 18:38:20 2020 -0700
Update README.md
commit c1a97a7767ccb2aa9afc7a5e72fd159e7c62ec02
Merge: 2bf86b8 f796708https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit 5a19011949398d06e744d8d5521ab4e6dfa06ab7
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 22:19:15 2020 +0800
Add assertion for <=2 gpus DDP
commit c8357ad5b15a0e6aeef4d7fe67ca9637f7322a4d
Merge: e742dd9 787582f
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 14 22:10:02 2020 +0800
Merge pull request #8 from MagicFrogSJTU/NanoCode012-patch-1
Modify number of dataloaders' workers
commit 787582f97251834f955ef05a77072b8c673a8397
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 20:38:58 2020 +0700
Fixed issue with single gpu not having world_size
commit 63648925288d63a21174a4dd28f92dbfebfeb75a
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 19:16:15 2020 +0700
Add assert message for clarification
Clarify why assertion was thrown to users
commit 69364d6050e048d0d8834e0f30ce84da3f6a13f3
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 17:36:48 2020 +0700
Changed number of workers check
commit d738487089e41c22b3b1cd73aa7c1c40320a6ebf
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 17:33:38 2020 +0700
Adding world_size
Reduce calls to torch.distributed. For use in create_dataloader.
commit e742dd9619d29306c7541821238d3d7cddcdc508
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 15:38:48 2020 +0800
Make SyncBN a choice
commit e90d4004387e6103fecad745f8cbc2edc918e906
Merge: 5bf8beb cd90360
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 14 15:32:10 2020 +0800
Merge pull request #6 from NanoCode012/patch-5
Update train.py
commit cd9036017e7f8bd519a8b62adab0f47ea67f4962
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 13:39:29 2020 +0700
Update train.py
Remove redundant `opt.` prefix.
commit 5bf8bebe8873afb18b762fe1f409aca116fac073
Merge: c9558a9 a1c8406https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit c9558a9b51547febb03d9c1ca42e2ef0fc15bb31
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 13:51:34 2020 +0800
Add device allocation for loss compute
commit 4f08c692fb5e943a89e0ee354ef6c80a50eeb28d
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:16:27 2020 +0800
Revert drop_last
commit 1dabe33a5a223b758cc761fc8741c6224205a34b
Merge: a1ce9b1 4b8450b
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:49 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit a1ce9b1e96b71d7fcb9d3e8143013eb8cebe5e27
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:21 2020 +0800
fix lr warning
commit 4b8450b46db76e5e58cd95df965d4736077cfb0e
Merge: b9a50ae 02c63ef
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Wed Jul 8 21:24:24 2020 +0800
Merge pull request #4 from NanoCode012/patch-4
Add drop_last for multi gpu
commit 02c63ef81cf98b28b10344fe2cce08a03b143941
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Wed Jul 8 10:08:30 2020 +0700
Add drop_last for multi gpu
commit b9a50aed48ab1536f94d49269977e2accd67748f
Merge: ec2dc6c 121d90bhttps://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit ec2dc6cc56de43ddff939e14c450672d0fbf9b3d
Merge: d0326e3 82a6182
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:34:31 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit d0326e398dfeeeac611ccc64198d4fe91b7aa969
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:31:24 2020 +0800
Add SyncBN
commit 82a6182b3ad0689a4432b631b438004e5acb3b74
Merge: 96fa40a 050b2a5
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 7 19:21:01 2020 +0800
Merge pull request #1 from NanoCode012/patch-2
Convert BatchNorm to SyncBatchNorm
commit 050b2a5a79a89c9405854d439a1f70f892139b1c
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:38:14 2020 +0700
Add cleanup for process_group
commit 2aa330139f3cc1237aeb3132245ed7e5d6da1683
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:07:40 2020 +0700
Remove apex.parallel. Use torch.nn.parallel
For future compatibility
commit 77c8e27e603bea9a69e7647587ca8d509dc1990d
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 01:54:39 2020 +0700
Convert BatchNorm to SyncBatchNorm
commit 96fa40a3a925e4ffd815fe329e1b5181ec92adc8
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 21:53:56 2020 +0800
Fix the datset inconsistency problem
commit 16e7c269d062c8d16c4d4ff70cc80fd87935dc95
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 11:34:03 2020 +0800
Add loss multiplication to preserver the single-process performance
commit e83805563065ffd2e38f85abe008fc662cc17909
Merge: 625bb49 3bdea3fhttps://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit 625bb49f4e52d781143fea0af36d14e5be8b040c
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 2 22:45:15 2020 +0800
DDP established
* Fixed destroy_process_group in DP mode
* Update torch_utils.py
* Update utils.py
Revert build_targets() to current master.
* Update datasets.py
* Fixed world_size attribute not found
Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-19 12:33:30 -07:00
aa08b2bd6f
test.py --task test updates
2020-07-15 18:25:34 -07:00
98f4a5af87
runs/exp increment bug fix #344
2020-07-15 16:07:40 -07:00
2377e5f6ee
FP16 EMA bug fix
2020-07-13 15:47:46 -07:00
140d84cca1
comment updates
2020-07-13 12:17:52 -07:00
8412d44dce
update datasets.py to add detect.py --source glob_command feature
2020-07-12 14:22:15 -07:00
eae33303d6
update datasets.py LoadImages() path improvements and Mixup
2020-07-12 14:14:51 -07:00
41ff83a98d
update torch_utils.py to FP16 EMA
2020-07-12 12:18:00 -07:00
f7a6d6c8d4
remove https:// prefix from curl commands fix #355
2020-07-11 13:00:24 -07:00
a586751904
multi-gpu ckpt filesize bug fix #253
2020-07-11 12:35:21 -07:00
d6f6b33a49
update datasets.py for image size check during caching
2020-07-11 00:21:52 -07:00
01a67a9742
Merge remote-tracking branch 'origin/master'
2020-07-10 20:36:26 -07:00
d3e786ed93
google_utils.py bug fix #355
2020-07-10 20:36:10 -07:00
41bb70b325
update create_pretrained()
2020-07-10 00:25:35 -07:00
1531795f5e
avoid git status check in Docker bug fix #342
2020-07-10 00:11:50 -07:00
1e9cf6a6e9
add file sizes to ckpt ops
2020-07-10 00:00:19 -07:00
59beae98e8
add if isfile() to get_hash()
2020-07-09 20:57:24 -07:00
6649770001
--rect test bug fix
2020-07-09 20:39:11 -07:00
520f5de6f0
Label caching foundational re-write #306
2020-07-09 20:07:16 -07:00
dd8e742ece
Merge pull request #334 from lorenzomammana/feature-multiple-datasets-training
...
Handle multiple datasets
2020-07-09 16:29:21 -07:00
dd33d2ab77
Update datasets.py
2020-07-09 16:28:20 -07:00
2110f5ee89
update log_dir to runs/exp #107
2020-07-09 16:03:10 -07:00
603ea0bfdc
update log_dir to runs/exp #107
2020-07-09 15:58:07 -07:00
24c5a941f0
--resume EMA fix #292
2020-07-09 15:09:06 -07:00
13a09fa68a
Update datasets.py
2020-07-09 13:47:20 -07:00
47bf1730fa
Update datasets.py
2020-07-09 13:45:55 -07:00
f9bab6b127
Update datasets.py
2020-07-09 13:36:23 -07:00
54a9e4f876
Refactor code to reduce duplication
2020-07-09 19:39:28 +02:00
94342acbe1
Handle multiple datasets
2020-07-09 11:52:12 +02:00
bf6f41567a
hyperparameter printout update
2020-07-08 17:21:00 -07:00
dc5e18390a
Merge branch 'master' into advanced_logging
2020-07-08 17:01:19 -07:00
cbe39a1dbb
Update utils.py
2020-07-08 16:14:32 -07:00
c654d18f00
Update utils.py
2020-07-08 16:11:22 -07:00
31b07b3550
Merge pull request #275 from wanghaoyang0106/master
...
[bug fix] potential problem if img fed to model is in rectangular shape
2020-07-08 00:07:54 -07:00
e8cf24b6c8
Initial model ensemble capability #318
2020-07-07 15:40:50 -07:00
95f0a56df7
Bug fix to get_latest_run when recent run is named
2020-07-07 11:19:49 -05:00
f517ba81c7
Merge branch 'master' into advanced_logging
2020-07-06 16:52:11 -05:00
121d90b3f2
update fuse_conv_and_bn()
2020-07-06 11:46:10 -07:00
38f5c1ad1d
pruning and sparsity initial commit
2020-07-05 13:41:21 -07:00
ccfa9373ae
Update to remove "module" attribute check
2020-07-04 23:12:57 +07:00
f83521ab9a
Fix save error for multi-gpu
2020-07-04 16:50:33 +07:00
ce0c58f678
update compute_loss()
2020-07-03 18:56:07 -07:00
af41083d52
EMA FP16 fix #279
2020-07-03 16:57:08 -07:00
f767023c56
offset and balance update
2020-07-03 14:07:00 -07:00
df224a0d8f
EMA bug fix #279
2020-07-03 11:56:14 -07:00
b57f83d005
[bug fix] potential problem if img fed to model is in rectangular shape
2020-07-03 13:09:21 +08:00
f02481c73a
Update torch_utils.py
2020-07-02 12:03:45 -07:00
fc7c42723d
Update utils.py
2020-07-02 12:01:43 -07:00
53cdaf6bf5
Update utils.py
2020-07-02 13:48:19 +08:00
9a9333d245
add replicate() to datasets.py
2020-06-29 17:10:33 -07:00
37e13f8846
update mosaic border
2020-06-27 13:50:15 -07:00
93a6765806
update mosaic border
2020-06-27 13:02:01 -07:00
e18e6811dc
Merge branch 'master' into advanced_logging
2020-06-27 10:13:03 -05:00
77fb8ee082
refactor dataloader
2020-06-26 19:30:09 -07:00
22fb2b0c25
refactor dataloader
2020-06-26 18:56:13 -07:00
b8557f87e3
add stride to datasets.py
2020-06-24 13:02:27 -07:00
3b062254a6
update utils.create_pretrained()
2020-06-24 12:53:43 -07:00
6c1b87a42e
update google_utils import
2020-06-22 23:00:23 -07:00
364fcfd7db
PANet update
2020-06-22 14:27:17 -07:00
fc171e2678
check_anchor_order() update
2020-06-21 14:08:59 -07:00
1f1917ef56
remove fast, add merge
2020-06-21 13:37:11 -07:00
e572bb0803
Add plot_results save location to log_dir
2020-06-21 09:36:28 -05:00
024a42f4fd
datasets.py vid_formats updated
2020-06-20 09:53:37 -07:00
5c2cd711df
vid_formats bug fix #146
2020-06-20 09:26:07 -07:00
10145833c3
strip_optimizer() default to save FP16 model
2020-06-18 00:16:47 -07:00
cce95e744d
backbone as FP16, save default to FP32
2020-06-18 00:13:18 -07:00
9fdb0fbacf
AutoAnchor bug fix # 117
2020-06-17 19:51:15 -07:00
945307beba
Add save_dir to plot_lr_scheduler and plot_labels
...
Set save_dir = log_dir in train.py
2020-06-17 16:03:18 -05:00
3263a204ea
Fix get_latest_run to search 'search_dir' recursivly
2020-06-17 10:34:37 -05:00
d34291733b
Fix get_latest_run() to search inside 'weights' subfolders
2020-06-17 10:26:55 -05:00
a557b7d924
check_img_size() update
2020-06-16 17:48:01 -07:00
034609414e
AutoAnchor fitness to screen
2020-06-16 17:08:45 -07:00
ef75f7313d
AutoAnchor n, imgsz bug fix
2020-06-16 16:48:01 -07:00
1dfc28527f
AutoAnchor BPR to 4 significant figures
2020-06-16 16:19:58 -07:00
4052603e1f
AutoAnchor update - improvement check
2020-06-16 15:54:20 -07:00
bafbc65ee3
AutoAnchor bug fix
2020-06-16 14:26:02 -07:00
25e51bcec7
add util function to get most recent last.pt file
...
added logic in train.py __main__ to handle resuming from a run
2020-06-16 15:50:27 -05:00
490f1e7b9c
add save_dir arg to plot_lr_scheduler, default to current dir.
...
Uncomment plot_lr_scheduler in train() and pass log_dir as save location
2020-06-16 15:13:03 -05:00
57a0ae3350
AutoAnchor implementation
2020-06-16 13:12:22 -07:00
95c46f7245
Merge pull request #79 from Lornatang/delete-redundant-thrid-party-library
...
Delete redundant thrid party library
2020-06-16 11:52:36 -07:00
ec81c7b5f2
check_anchors() bug fix #90
2020-06-16 10:14:04 -07:00
8b26e89006
AutoAnchor bug fix #72
2020-06-16 00:53:34 -07:00
8fa3724072
kmean_anchors() update
2020-06-16 00:40:21 -07:00
1c0b6236e3
update fast mode
2020-06-16 00:11:29 -07:00
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.
2020-06-16 08:28:52 +08:00
db2c3acd3a
updated testing settings, rebalanced towards FP16 latency
2020-06-15 16:55:57 -07:00
bd3e389130
update google_utils.py
2020-06-15 14:39:04 -07:00
c5966abba8
glob search bug fix #77
2020-06-15 12:08:57 -07:00
54795a4f0a
fix problem
2020-06-15 16:11:06 +08:00
260b1729f0
FP16 inference update
2020-06-14 18:32:29 -07:00
93bbf1b87a
best possible recall fix
2020-06-13 15:25:53 -07:00
31f3310029
assert best possible recall > 0.9 before training
2020-06-13 15:05:41 -07:00
099e6f5ebd
--img-size stride-multiple verification
2020-06-12 22:10:46 -07:00
3a5c5328c5
ONNX, BCEBlurWithLogitsLoss, plot_study updates
2020-06-11 14:44:09 -07:00
ac93e9c811
Fix Google Drive URI for YOLOv5 weights
2020-06-10 14:26:10 -07:00
c14368d768
new CSP model release
2020-06-09 14:31:04 -07:00
ff02ae0869
module updates
2020-06-08 22:13:01 -07:00
1f698af6f1
create_backbone with SourceChangeWarning update
2020-06-07 15:45:05 -07:00
8fe299f179
model fuse
2020-06-07 14:10:30 -07:00
b810b21299
augmented inference
2020-06-05 21:14:15 -07:00
c12023f145
updates
2020-06-05 20:50:43 -07:00
66d73e4cf3
NMS updates
2020-06-05 18:44:39 -07:00
948bcdd219
--classes bug fix #17
2020-06-05 13:29:26 -07:00
22d6088205
speed-reproducibility fix #17
2020-06-05 13:07:09 -07:00
0362f103ef
updates
2020-06-04 10:19:06 -07:00
2a0aff685a
dataset padding mode
2020-06-03 13:55:33 -07:00
eb97b2e413
NMS fast mode
2020-06-03 13:02:59 -07:00
ee8988b8a2
datasets.py self.batch_shapes fix
2020-06-01 18:11:18 -07:00
f96e8f3edf
updates
2020-05-31 19:51:32 -07:00
b10609fe6b
updates
2020-05-31 15:07:04 -07:00
72eed215ff
updates
2020-05-31 09:25:42 -07:00
5a79b5f65c
updates
2020-05-30 15:22:09 -07:00
4f44aaf26b
updates
2020-05-30 15:18:59 -07:00
1e84a23f38
initial commit
2020-05-29 17:04:54 -07:00
Glenn Jocher
Ayush Chaurasia
ketan-b
Valentin Aliferov
Feras Oughali
Zigarss
Yonghye Kwon
Mai Thanh Minh
xiaowk5516
dependabot[bot]
Dean Mark
chocosaj
tudoulei
yeric1789
Charles Frye
Adrian Holovaty
NanoCode012
r-blmnr
Michael Heilig
JoshSong
Burhan
Robin
Ben Milanko
Phat Tran
zzttqu
maxupp
Yann Defretin
Ryan Avery
Iden Craven
xiaowo1996
Transigent
train255
Abhiram V
Tommy in Tongji
Polydefkis Gkagkos
Hu Ye
yxNONG
igornishka
yujun
Jacobsolawetz
Shiwei Song
Jirka Borovec
Jiacong Fang
haoyuefan
Guilhen
tkianai
rafale77
Willie Maddox
Francisco Ingham
Quan Lin
Hatovix
Marc
Liu Changyu
AlexWang1900
YuriRibeiro
pritul dave
yzchen
lorenzomammana
Alex Stoken
wanghaoyang0106
Lornatang
Trevor Sullivan
