a3ecf0fd64
Anchor override ( #2350 )
2021-03-02 23:08:21 -08:00
fab5085674
EMA bug fix 2 ( #2330 )
...
* EMA bug fix 2
* update
2021-03-01 17:13:34 -08:00
fd96810518
remove TTA 1 pixel offset ( #2325 )
2021-02-28 21:14:08 -08:00
cbd55da5d2
Update yolo.py ( #2120 )
...
* Avoid mutable state in Detect
* LoadImages() pathlib update (#2140 )
* 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>
* Update train.py test batch_size (#2148 )
* Update train.py
* Update loss.py
* Update train.py (#2149 )
* Linear LR scheduler option (#2150 )
* Linear LR scheduler option
* Update train.py
* Update data-autodownload background tasks (#2154 )
* Update get_coco.sh
* Update get_voc.sh
* Update detect.py (#2167 )
Without this cv2.imshow opens a window but nothing is visible
* Update requirements.txt (#2173 )
* 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.
* Changed socket port and added timeout (#2176 )
* PyTorch Hub results.save('path/to/dir') (#2179 )
* 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
* Created using Colaboratory
* 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>
* 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>
* LoadStreams() frame loss bug fix (#2222 )
* Update yolo.py channel array (#2223 )
* Add check_imshow() (#2231 )
* Add check_imshow()
* Update general.py
* Update general.py
* Update CI badge (#2230 )
* Add isdocker() (#2232 )
* Add isdocker()
* Update general.py
* Update general.py
* YOLOv5 Hub URL inference bug fix (#2250 )
* Update common.py
* Update common.py
* Update common.py
* Improved hubconf.py CI tests (#2251 )
* Unified hub and detect.py box and labels plotting (#2243 )
* reset head
* Update inference default to multi_label=False (#2252 )
* Update inference default to multi_label=False
* bug fix
* Update plots.py
* Update plots.py
* Robust objectness loss balancing (#2256 )
* Created using Colaboratory
* Update minimum stride to 32 (#2266 )
* 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>
* Update greetings.yml for auto-rebase on PR (#2272 )
* Update Dockerfile with apt install zip (#2274 )
* FLOPS min stride 32 (#2276 )
Signed-off-by: xiaowo1996 <429740343@qq.com>
* Update README.md
* Amazon AWS EC2 startup and re-startup scripts (#2185 )
* Amazon AWS EC2 startup and re-startup scripts
* Create resume.py
* cleanup
* Amazon AWS EC2 startup and re-startup scripts (#2282 )
* Update train.py (#2290 )
* Update train.py
* Update train.py
* Update train.py
* Update train.py
* Create train.py
* 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
* Improved model+EMA checkpointing 2 (#2295 )
* Fix labels being missed when image extension appears twice in filename (#2300 )
* W&B entity support (#2298 )
* W&B entity support
* shorten wandb_entity to entity
Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Avoid mutable state in Detect
* Update yolo and remove .to(device)
Co-authored-by: Oleg Boiko <oboiko@chegg.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: train255 <thanhdd.it@gmail.com>
Co-authored-by: ab-101 <56578530+ab-101@users.noreply.github.com>
Co-authored-by: Transigent <wbdsmith@optusnet.com.au>
Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
Co-authored-by: Daniel Khromov <danielkhromov@gmail.com>
Co-authored-by: VdLMV <jaap@vdlmv.nl>
Co-authored-by: Jaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
Co-authored-by: Yann Defretin <kinoute@gmail.com>
Co-authored-by: Aditya Lohia <64709773+aditya-dl@users.noreply.github.com>
Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: xiaowo1996 <429740343@qq.com>
Co-authored-by: Iden Craven <iden.craven@gmail.com>
Co-authored-by: Jan Hajek <toretak@users.noreply.github.com>
Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
2021-02-26 15:07:40 -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
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
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
f8464b4f66
Update yolo.py channel array ( #2223 )
2021-02-15 21:21:53 -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
404749a33c
PyTorch Hub results.save('path/to/dir') ( #2179 )
2021-02-10 16:10:43 -08:00
9646ca438a
Update plot_study() ( #2112 )
2021-02-01 13:51:11 -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
08d3119e09
Add histogram equalization fcn ( #2049 )
2021-01-26 17:30:42 -08:00
2fc4760257
Update autoshape .print() and .save() ( #2022 )
2021-01-22 16:38:06 -08:00
aac33f87ad
Update inference multiple-counting ( #2019 )
...
* Update inference multiple-counting
* update github check
2021-01-22 15:39:08 -08:00
4501169547
Update yolo.py with yaml.SafeLoader ( #1970 )
2021-01-18 10:47:00 -08:00
03ebe6ea14
Add nn.SiLU inplace in attempt_load() ( #1940 )
...
* sleep 3 sec to load data .
* Update train.py
* Add nn.SiLU inplace in attempt_load()
Co-authored-by: wangliang <wangliang@codingapi.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-01-15 09:58:41 -08:00
1d1c0567a4
PyTorch Hub results.render() ( #1897 )
2021-01-10 20:06:11 -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
0b6266f5e0
Update yolov3-tiny.yaml ( #1823 )
2021-01-02 13:01:05 -08:00
14b0abe2d0
autoShape() default for PyTorch Hub models ( #1692 )
...
* Add autoshape parameter
* Remove autoshape call in ReadMe
* Update hubconf.py
* file/URI inputs and autoshape check passthrough
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-12-26 19:58:26 -08:00
9922c1183f
leaf Variable inplace bug fix ( #1759 )
2020-12-22 17:27:40 -08:00
394d1c89f3
Input channel yaml['ch'] addition ( #1741 )
2020-12-19 10:54:01 -08:00
6bd5e8bca7
nn.SiLU() export support ( #1713 )
2020-12-16 17:55:57 -08:00
799724108f
Update C3 module ( #1705 )
2020-12-15 22:13:08 -08:00
fa8f1fb0e9
Simplify autoshape() post-process ( #1653 )
...
* Simplify autoshape() post-process
* cleanup
* cleanup
2020-12-09 07:44:06 -08:00
791dadb51c
Pycocotools best.pt after COCO train ( #1616 )
...
* Pycocotools best.pt after COCO train
* cleanup
2020-12-06 14:58:33 +01:00
d929bb656c
Implement default class names ( #1609 )
2020-12-06 12:41:37 +01:00
ba48f867ea
Add bias to Classify() ( #1601 )
2020-12-04 15:06:33 +01:00
199c9c7874
Detection() device bug fix ( #1455 )
2020-11-19 13:45:36 +01:00
225845e781
Update test.py for IoU in native image-space ( #1439 )
...
* Update test.py for IoU in native image-space
* remove redundant
* gn to device
* remove output scale_coords
* --img-size correction
* update
* native-space labels
* pred to predn
* remove clip_coords()
2020-11-18 22:50:21 +01:00
97c6a59dd1
PANet anchor order correction ( #1420 )
2020-11-17 10:13:00 +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
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
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
15a10609fe
autoshape() update for PIL greyscale inputs ( #1279 )
...
* autoshape update for PIL greyscale inputs
* autoshape update for PIL greyscale inputs
2020-11-04 14:20:11 +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
f532bc68a9
GhostConv() bug fix ( #1176 )
2020-10-20 13:42:11 +02:00
3b57cb5641
Simplified inference ( #1153 )
2020-10-15 20:10:08 +02:00
d45e349a1e
Rearrange export input after checks ( #1118 )
...
img size checks are warnings rather than errors, so current implementation allows improperly formed model inputs.
2020-10-11 16:23:36 +02:00
c2403d785c
fuse() bug fix
2020-10-11 00:42:54 +02:00
883924d9dc
classifier, export, torch seed updates
2020-10-06 14:54:02 +02:00
a12698f4c9
Update ci-testing.yml reflecting sys.path additions ( #869 )
...
* Update ci-testing.yml
* Update yolo.py
* Update ci-testing.yml
* Update export.py
* Update yolo.py
2020-10-04 18:50:32 +02:00
d55c326dda
export.py folder sys.path addition ( #1052 )
2020-10-04 17:42:05 +02:00
4fce0096fa
model.add_nms() method
2020-09-20 16:31:40 -07:00
2f77cf33f6
.fuse() additional error checking
2020-09-20 12:04:20 -07:00
c4cb78570c
add NMS to pretrained pytorch hub models
2020-09-18 18:17:11 -07:00
6062bb7dee
print self.yaml['nc'] override update
2020-09-11 17:44:11 -07:00
35dc935504
remove `nn.Conv2d` in default settings ( #949 )
2020-09-11 16:35:48 -07:00
d68179953d
Update export.py
2020-09-02 13:23:29 -07:00
1e7674d4dd
Update export.py --img-size comment
2020-08-25 23:07:22 -07:00
4fb8cb353f
Update export.py with v3.0 Hardswish() support ( #831 )
2020-08-24 21:59:26 -07:00
4d7f222f73
Update export.py with v3.0 Hardswish() support
2020-08-24 21:47:49 -07:00
a8751e50de
model.fuse() fix for export.py ( #827 )
2020-08-24 19:27:54 -07:00
d0f98c0095
Update yolo.py with model_info(verbose=False)
2020-08-21 16:20:45 -07:00
b42e8a531b
optimize imports
2020-08-20 21:17:40 -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
0892c44bc4
Fix Logging ( #719 )
...
* Add logging setup
* Fix fusing layers message
* Fix logging does not have end
* Add logging
* Change logging to use logger
* Update yolo.py
I tried this in a cloned branch, and everything seems to work fine
* Update yolo.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-12 14:18:19 -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
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
2020-08-02 15:47:36 -07:00
d989bc9260
remove NBSP
2020-08-02 14:23:05 -07:00
2268f9c44f
update *.yaml output layer comments
2020-07-30 11:31:47 -07:00
adaf1d8ac1
Update models/hub *.yaml files for v2.0 ( #546 )
...
* fix #543 issue
* Update yolov3-spp.yaml
* Update yolov5-fpn.yaml
* Update yolov5-panet.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-29 14:02:07 -07:00
43a616a955
PyTorch 1.6.0 compatability updates
2020-07-28 15:38: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
9da56b62dd
v2.0 Release ( #491 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-23 15:34:23 -07:00
a97c3f94ec
update common.py Classify()
2020-07-16 23:59:51 -07:00
5387d4747d
update common.py add Classify()
2020-07-16 17:18:41 -07:00
2b1892430f
yolo.py bug fix
2020-07-13 15:21:19 -07:00
02445d176d
improved model.yaml source tracking
2020-07-13 14:35:47 -07:00
9006b85d21
update export.py torchscript file extension
2020-07-10 11:56:01 -07:00
16f6834486
update train.py and experimental.py
2020-07-08 14:23:34 -07:00
a9918fbf2b
updated coreml conversion to do pixel scaling between 0.0 and 1.0
2020-07-08 09:43:33 +03:00
e8cf24b6c8
Initial model ensemble capability #318
2020-07-07 15:40:50 -07:00
04bdbe4104
fuse update
2020-07-05 23:16:50 -07:00
5ba1de0cdc
update experimental.py with Ensemble() module
2020-07-05 15:02:56 -07:00
38f5c1ad1d
pruning and sparsity initial commit
2020-07-05 13:41:21 -07:00
997ba7b346
import yaml in yolo.py
2020-07-05 12:50:04 -07:00
762b06de4b
Merge remote-tracking branch 'origin/master'
2020-07-04 23:07:32 -07:00
4e2d24602d
update yolo.py
2020-07-04 23:07:26 -07:00
bb3c346916
model.yaml nc inherited from dataset.yaml
2020-07-04 17:51:54 -07:00
659ad748c8
update get_voc.sh
2020-07-04 17:13:43 -07:00
7bd7b2c090
update export.py
2020-07-03 20:05:50 -07:00
655895a838
module updates
2020-07-03 13:46:12 -07:00
a9d20eba3e
export.py update
2020-07-03 11:50:59 -07:00
569fbc63c8
export.py update
2020-07-03 11:29:53 -07:00
a03adb5ab8
Merge remote-tracking branch 'origin/master'
2020-07-01 19:16:06 -07:00
1fca7a7f24
autopad() update in common.py
2020-07-01 19:15:59 -07:00
a62a1c2c67
export.py update
2020-07-01 16:14:49 -07:00
f1d67f4110
update export.py
2020-07-01 15:46:15 -07:00
b5659d1195
module updates
2020-07-01 11:44:49 -07:00
615d6d0cfa
FROM nvcr.io/nvidia/pytorch:20.06-py3
2020-06-29 14:00:13 -07:00
e8bf385795
Merge pull request #223 from ArmyOfRobots/master
...
Adding torchscript export
2020-06-29 13:39:53 -07:00
0bc80e1e57
onnx_export.py bug fix #224
2020-06-28 17:07:03 -07:00
685091a3fc
Adding torchscript export
2020-06-28 13:17:05 -07:00
Glenn Jocher
oleg
Aditya Lohia
Yann Defretin
VdLMV
Daniel Khromov
Abhiram V
lorne
NanoCode012
Yi_Zhang
PetrDvoracek
tomguluson92
Jirka Borovec
Liu Changyu
Laurentiu Diaconu
Jake Poznanski