yolov5

Commit Graph

Author	SHA1	Message	Date
Glenn Jocher	1f8d716ec9	yolo.py header (#3347 )	3 years ago
Glenn Jocher	aad99b63d6	TensorBoard DP/DDP graph fix (#3325 )	3 years ago
Glenn Jocher	407dc5008e	Update README.md (#3320 )	3 years ago
Glenn Jocher	61ea23c3fe	Implement `@torch.no_grad()` decorator (#3312 ) * `@torch.no_grad()` decorator * Update detect.py	3 years ago
Glenn Jocher	73a92dc1b6	Explicit `git clone` master (#3311 )	3 years ago
Glenn Jocher	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.	3 years ago
Glenn Jocher	ee24ae11ea	Comment Cython (#3297 ) Ultralytics REST API fix.	3 years ago
yeric1789	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>	3 years ago
Glenn Jocher	0e2f2cbb51	Update LoadStreams init fallbacks (#3295 )	3 years ago
Charles Frye	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	3 years ago
Glenn Jocher	dd7f0b7e05	Fix TypeError: 'PosixPath' object is not iterable (#3285 )	3 years ago
Glenn Jocher	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	3 years ago
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

1 2 3 4 5 ...

1241 Commits (b3e2f4e08d6a16bf153c9b56bbc0001a52dd24e3) All Branches Search

1241 Commits (b3e2f4e08d6a16bf153c9b56bbc0001a52dd24e3)

All Branches