yolov5

Commit Graph

Author	SHA1	Message	Date
Glenn Jocher	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.	3 years ago
Glenn Jocher	10d56d784e	Assert `--image-weights` not combined with DDP (#3275 )	3 years ago
Glenn Jocher	7b36e38cf8	Check CoreML models.train() mode (#3262 ) * Check CoreML models.train() mode * Update export.py	3 years ago
Glenn Jocher	7b863a34dc	Add Cython (#3217 ) Cython required by pycocotools in certain cases, i.e. https://github.com/cocodataset/cocoapi/issues/172	3 years ago
yeric1789	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>	3 years ago
Glenn Jocher	b7cd1f540d	TensorBoard add_graph() fix (#3236 )	3 years ago
Adrian Holovaty	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>	3 years ago
Glenn Jocher	13a1c72699	Update datasets.py (#3216 )	3 years ago
Glenn Jocher	ffb47ffbeb	Silent Patches Bug Fix (#3214 )	3 years ago
Glenn Jocher	36b0a9e7d0	Add CAP_PROP_FRAME_COUNT for YouTube sources (#3193 )	3 years ago
Glenn Jocher	ae53f50ae7	Plot labels histogram colors (#3192 )	3 years ago
Christoph Gerum	b133baa336	Add `device` argument to PyTorch Hub models (#3104 ) * Allow to manual selection of device for torchhub models * single line device nested torch.device(torch.device(device)) ok Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	3 years ago
Cristi Fati	9ab561dbfc	Parameterize ONNX `--opset-version` (#3154 )	3 years ago
Yonghye Kwon	be86c21c73	rename class autoShape -> AutoShape (#3173 ) * rename class autoShape -> AutoShape follow other class naming convention * rename class autoShape -> AutoShape follow other classes' naming convention * rename class autoShape -> AutoShape	3 years ago
Glenn Jocher	17b0f71538	SKU-110K CVPR2019 Dataset Auto-Download (#3167 ) * SKU-110K CVPR2019 Dataset Auto-Download This PR adds the SKU-110K CVPR2019 Dataset https://github.com/eg4000/SKU110K_CVPR19 to the list of YOLOv5 🚀 auto-download list. - [x] [COCO](https://cocodataset.org/#home) - [x] [COCO128](https://www.kaggle.com/ultralytics/coco128) - [x] [VOC](http://host.robots.ox.ac.uk/pascal/VOC/) - [x] [Argoverse-HD](http://www.cs.cmu.edu/~mengtial/proj/streaming/) - [x] [VisDrone2019-DET](https://github.com/VisDrone/VisDrone-Dataset) - [x] [GlobalWheat-2020](http://www.global-wheat.com/) - [ ] [KITTI](https://www.kaggle.com/twaldo/kitti-object-detection) - [x] [SKU110](https://github.com/eg4000/SKU110K_CVPR19) (grocery store items) - [ ] [CityScapes](https://www.cityscapes-dataset.com/) - [x] [Objects365](https://www.objects365.org/overview.html) - [ ] [OpenImages](https://storage.googleapis.com/openimages/web/index.html) ```yaml # SKU-110K retail items dataset https://github.com/eg4000/SKU110K_CVPR19 # Train command: python train.py --data SKU-110K.yaml # Default dataset location is next to YOLOv5: # /parent_folder # /datasets/SKU-110K # /yolov5 ``` * Update SKU-110K.yaml * Update SKU-110K.yaml	3 years ago
Glenn Jocher	06372b1465	Scope all hubconf.py imports for torch.hub.list() (#3145 ) * Scope all hubconf.py imports for torch.hub.list() * Update hubconf.py	3 years ago
Glenn Jocher	1935266951	Scope imports for torch.hub.list() improvement (#3144 )	3 years ago
Glenn Jocher	b098600763	Update requirements.txt `onnx>=1.9.0` (#3143 )	3 years ago
Glenn Jocher	f12cef85a3	TorchScript, ONNX, CoreML Export tutorial title (#3142 )	3 years ago
Cristi Fati	d9b4e6b748	Add `--include torchscript onnx coreml` argument (#3137 ) * Allow users to skip exporting in formats that they don't care about * Correct comments * Update export.py renamed --skip-format to --exclude * Switched format from exclude to include (as instructed by @glenn-jocher) * cleanup Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	3 years ago
KC-Zhang	bd6f6a784a	Add yolov5/__init__.py (#3127 )	3 years ago
Glenn Jocher	7918eed040	Update https://ultralytics.com/images/zidane.jpg (#3140 )	3 years ago
Glenn Jocher	c1c7eb023f	Update JSON response (#3139 )	3 years ago
Glenn Jocher	52c05707d6	Update Colors() (#3046 ) * Update Colors() * update colors * update colors	3 years ago
Glenn Jocher	abfcf9eb79	Replace print() with logging.info() in trainloader (#3103 ) Might indirectly help #3095 by providing better visibility on source of corruption.	3 years ago
Glenn Jocher	25f8ab835e	detect.py streaming source `--save-crop` bug fix (#3102 ) * detect.py streaming source --save-crop bug fix Possible fix for #3100. * () parenthesis	3 years ago
Glenn Jocher	d6a71ae5f2	Update GlobalWheat2020.yaml test: # 1276 images (#3101 )	3 years ago
Glenn Jocher	60fe54449d	Update train.py (#3099 )	3 years ago
Glenn Jocher	525f4f86a9	Add --optimize argument (#3093 ) Fix for c++ runtime errors in https://github.com/ultralytics/yolov5/issues/2973	3 years ago
Glenn Jocher	57b0d3a605	Add check_python() (#3088 ) * Add check_python() Checks python version against minimum version of 3.7.0. * remove packaging dependency * refactor import	3 years ago
Glenn Jocher	91547edec1	Update detect.py (#3087 ) * Update detect.py * Update detect.py	3 years ago
Glenn Jocher	251aeafcb1	Update P5 + P6 model ensembling (#3082 )	3 years ago
Ayush Chaurasia	d2a17289c9	Explicitly convert artifact path to posix_path (#3067 ) * Explicitly convert artifact path to posix_path * Remove redudant str() casting	3 years ago
Glenn Jocher	e97d129db4	Update export.py with --train mode argument (#3066 )	3 years ago
kepler62f	f2de1ad2aa	Comment fix (#3058 ) Co-authored-by: kepler62f <>	3 years ago
Glenn Jocher	3ef3a95cfa	Do not optimize CoreML TorchScript model (#3055 )	3 years ago
Jiacong Fang	b8be76f915	Add get_coco128.sh for downloading the coco128 dataset (#3047 ) * Add get_coco128.sh * Update get_coco128.sh Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	3 years ago
Ayush Chaurasia	b18ca31ff5	use check_file for --data (#3035 )	3 years ago
Glenn Jocher	85b6fdda0f	FROM nvcr.io/nvidia/pytorch:21.03-py3 Issues with 21.04	3 years ago
Glenn Jocher	4c7d686d7e	FROM nvcr.io/nvidia/pytorch:21.04-py3 (#3030 )	3 years ago
Glenn Jocher	8cab907f60	download() ThreadPool update (#3027 ) * download() ThreadPool update * update train image count * cid + 1	3 years ago
Glenn Jocher	5189b3addb	Objects365 Update	3 years ago
jylink	b292837e36	Fix ONNX export using --grid --simplify --dynamic simultaneously (#2982 ) * Update yolo.py * Update export.py * fix export grid * Update export.py, remove detect export attribute * rearrange if order * remove --grid, default inplace=False * rename exp_dynamic to onnx_dynamic, comment * replace bs with 1 in anchor_grid[i] index 0 * Update export.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	3 years ago
Glenn Jocher	41cc7caee6	Add NMS threshold checks (#3020 ) * Add NMS threshold checks * fix	3 years ago
Glenn Jocher	fcd57028d2	Add is_colab() function (#3018 )	3 years ago
Hodovo	e2a80c6c0f	Add support for FP16 (half) to export.py (#3010 ) * Added support for fp16 (half) to export.py * minimize code additions Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	3 years ago
Glenn Jocher	31ee54c28c	Hubconf.py bug fix (#3007 )	3 years ago
Glenn Jocher	7c89c829e3	cleanup	3 years ago
Glenn Jocher	c1a44ed9c5	Update hubconf.py for unified loading (#3005 )	3 years ago
Glenn Jocher	37eaffec7d	Curl update (#3004 ) * Curl update * Curl update	3 years ago

... 3 4 5 6 7 ...

1329 Commits (4bad9147611238f31a66ba5414b35e8ca604ea37) All Branches Search

1329 Commits (4bad9147611238f31a66ba5414b35e8ca604ea37)

All Branches