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
ca5b10b759
Update train.py ( #2290 )
...
* Update train.py
* Update train.py
* Update train.py
* Update train.py
* Create train.py
2021-02-24 13:31:20 -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
7a6870b81f
Update README.md
2021-02-23 11:27:44 -08:00
83dc1b4484
FLOPS min stride 32 ( #2276 )
...
Signed-off-by: xiaowo1996 <429740343@qq.com>
2021-02-23 09:38:56 -08:00
cc79f3a9ea
Update Dockerfile with apt install zip ( #2274 )
2021-02-22 22:50:00 -08:00
32dd1614f4
Update greetings.yml for auto-rebase on PR ( #2272 )
2021-02-22 18:34:07 -08:00
95aefea493
Dynamic ONNX engine generation ( #2208 )
...
* add: dynamic onnx export
* delete: test onnx inference
* fix dynamic output axis
* Code reduction
* fix: dynamic output axes, dynamic input naming
* Remove fixed axes
Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-02-21 21:50:44 -08:00
e27ca0d845
Update minimum stride to 32 ( #2266 )
2021-02-21 21:46:42 -08:00
095d2c11d8
Created using Colaboratory
2021-02-20 13:21:58 -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
ab2da5edc8
Merge remote-tracking branch 'origin/master'
2021-02-19 15:21:04 -08:00
47faf95079
reset head
2021-02-19 15:20:41 -08:00
5f42643a53
Unified hub and detect.py box and labels plotting ( #2243 )
2021-02-19 12:38:05 -08:00
db28ce61ac
Improved hubconf.py CI tests ( #2251 )
2021-02-19 12:35:38 -08:00
9d87307784
YOLOv5 Hub URL inference bug fix ( #2250 )
...
* Update common.py
* Update common.py
* Update common.py
2021-02-19 11:22:42 -08:00
d2e754b67b
Add isdocker() ( #2232 )
...
* Add isdocker()
* Update general.py
* Update general.py
2021-02-16 15:58:07 -08:00
5a40ce65ce
Update CI badge ( #2230 )
2021-02-16 15:27:24 -08:00
26c2e54c8f
Add check_imshow() ( #2231 )
...
* Add check_imshow()
* Update general.py
* Update general.py
2021-02-16 13:56:47 -08:00
f8464b4f66
Update yolo.py channel array ( #2223 )
2021-02-15 21:21:53 -08:00
7b833e37bf
LoadStreams() frame loss bug fix ( #2222 )
2021-02-15 11:02:20 -08:00
3ff783c18f
TTA augument boxes one pixel shifted in de-flip ud and lr ( #2219 )
...
* TTA augument boxes one pixel shifted in de-flip ud and lr
* PEP8 reformat
Co-authored-by: Jaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-02-15 10:49:22 -08:00
3e560e2fae
YOLOv5 PyTorch Hub results.save() method retains filenames ( #2194 )
...
* save results with name
* debug
* save original imgs names
* Update common.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-02-12 15:37:51 -08:00
17ac94b796
Created using Colaboratory
2021-02-11 22:39:37 -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
404749a33c
PyTorch Hub results.save('path/to/dir') ( #2179 )
2021-02-10 16:10:43 -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
ace3e02e40
Update requirements.txt ( #2173 )
2021-02-09 22:03:29 -08:00
c32b0aff76
Update detect.py ( #2167 )
...
Without this cv2.imshow opens a window but nothing is visible
2021-02-08 23:13:40 -08:00
a5359f6c32
Update data-autodownload background tasks ( #2154 )
...
* Update get_coco.sh
* Update get_voc.sh
2021-02-06 22:13:39 -08:00
6b634c6b87
Linear LR scheduler option ( #2150 )
...
* Linear LR scheduler option
* Update train.py
2021-02-06 11:26:54 -08:00
ad839eda38
Update train.py ( #2149 )
2021-02-06 11:21:04 -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
aa02b94825
Add Kaggle badge ( #2090 )
2021-01-30 11:51:16 -08:00
eeb2bbf648
Add Kaggle badge ( #2090 )
...
* Update README.md
* Update greetings.yml
* Created using Colaboratory
2021-01-30 11:48:03 -08:00
170d12e506
Add Amazon Deep Learning AMI environment ( #2085 )
...
* Update greetings.yml
* Update README.md
2021-01-29 13:49:00 -08:00
64628d6fc5
Update tutorial.ipynb
2021-01-29 13:40:03 -08:00
6e6f77be47
Add YOLOv5-P6 models ( #2083 )
2021-01-29 11:57:29 -08:00
6bfa9c2422
GhostConv update ( #2082 )
2021-01-29 11:25:01 -08:00
2acbe9699a
W&B epoch logging update ( #2073 )
2021-01-28 18:57:24 -08:00
630ec06af6
Update to colors.TABLEAU_COLORS ( #2069 )
2021-01-28 12:37:21 -08:00
Glenn Jocher
xiaowo1996
Aditya Lohia
Yann Defretin
VdLMV
Daniel Khromov
NanoCode012
Transigent
ab-101
train255
Ayush Chaurasia