yolov5

Commit Graph

Author	SHA1	Message	Date
Glenn Jocher	6e8c5b7678	Improve git_describe() (#2633 ) Catch 'fatal: not a git repository' returns and return '' instead (observed in GCP Hub checks).	3 years ago
Glenn Jocher	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 ... ```	3 years ago
maxupp	196bf10603	Add '*.mpo' to supported image formats (#2615 ) Co-authored-by: Max Uppenkamp <max.uppenkamp@inform-software.com>	3 years ago
Glenn Jocher	77415a42e5	Update git_describe() for remote dir usage (#2606 )	3 years ago
Glenn Jocher	3bb414890a	resume.py typo (#2603 )	3 years ago
Glenn Jocher	d4456e43b2	Update segment2box() comment (#2600 )	3 years ago
Glenn Jocher	333ccc5b0f	YOLOv5 PyTorch Hub models >> check_requirements() (#2592 ) Improved user-feedback following requirements auto-update.	3 years ago
Glenn Jocher	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.	3 years ago
Glenn Jocher	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	3 years ago
Ayush Chaurasia	1bf9365280	W&B DDP fix (#2574 )	3 years ago
Ayush Chaurasia	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>	3 years ago
Glenn Jocher	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	3 years ago
Glenn Jocher	2d41e70e82	Scipy kmeans-robust autoanchor update (#2470 ) Fix for https://github.com/ultralytics/yolov5/issues/2394	3 years ago
Yann Defretin	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>	3 years ago
Glenn Jocher	08d4918d7f	labels.jpg class names (#2454 ) * labels.png class names * fontsize=10	3 years ago
Glenn Jocher	f01f3223d5	Integer printout (#2450 ) * Integer printout * test.py 'Labels' * Update train.py	3 years ago
Glenn Jocher	e8a2b83268	GCP sudo docker userdata.sh (#2393 ) * GCP sudo docker * cleanup	3 years ago
Glenn Jocher	7c2c95732c	AWS wait && echo "All tasks done." (#2391 )	3 years ago
Glenn Jocher	ba18528b47	bbox_iou() stability and speed improvements (#2385 )	3 years ago
Yonghye Kwon	cd8ed3521d	image weights compatible faster random index generator v2 for mosaic augmentation (#2383 ) image weights compatible faster random index generator v2 for mosaic augmentation	3 years ago
Ryan Avery	2c56ad5436	Confusion matrix background axis swap (#2114 )	3 years ago
Glenn Jocher	fab5085674	EMA bug fix 2 (#2330 ) * EMA bug fix 2 * update	3 years ago
Iden Craven	a82dce7faa	Fix labels being missed when image extension appears twice in filename (#2300 )	3 years ago
Glenn Jocher	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	3 years ago
Glenn Jocher	0070995bd5	Amazon AWS EC2 startup and re-startup scripts (#2282 )	3 years ago
Glenn Jocher	d5d275b6e9	Amazon AWS EC2 startup and re-startup scripts (#2185 ) * Amazon AWS EC2 startup and re-startup scripts * Create resume.py * cleanup	3 years ago
xiaowo1996	83dc1b4484	FLOPS min stride 32 (#2276 ) Signed-off-by: xiaowo1996 <429740343@qq.com>	3 years ago
Glenn Jocher	6f5d6fcdaa	Robust objectness loss balancing (#2256 )	3 years ago
Glenn Jocher	c09964c27c	Update inference default to multi_label=False (#2252 ) * Update inference default to multi_label=False * bug fix * Update plots.py * Update plots.py	3 years ago
Glenn Jocher	47faf95079	reset head	3 years ago
Glenn Jocher	d2e754b67b	Add isdocker() (#2232 ) * Add isdocker() * Update general.py * Update general.py	3 years ago
Glenn Jocher	26c2e54c8f	Add check_imshow() (#2231 ) * Add check_imshow() * Update general.py * Update general.py	3 years ago
Glenn Jocher	7b833e37bf	LoadStreams() frame loss bug fix (#2222 )	3 years ago
Glenn Jocher	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	3 years ago
NanoCode012	a5d5f9262d	Changed socket port and added timeout (#2176 )	3 years ago
Transigent	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.	3 years ago
Glenn Jocher	86897e3663	Update train.py test batch_size (#2148 ) * Update train.py * Update loss.py	3 years ago
train255	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>	3 years ago
Glenn Jocher	4bdc5a397e	LoadImages() pathlib update (#2140 )	3 years ago
Ayush Chaurasia	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>	3 years ago
Glenn Jocher	9646ca438a	Update plot_study() (#2112 )	3 years ago
Glenn Jocher	be9edffded	Update test.py --task speed and study (#2099 ) * Add --speed benchmark * test range 256 - 1536 * update * update * update * update	3 years ago
Glenn Jocher	a18efc3a73	Add variable-stride inference support (#2091 )	3 years ago
Glenn Jocher	630ec06af6	Update to colors.TABLEAU_COLORS (#2069 )	3 years ago
Glenn Jocher	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	3 years ago
Glenn Jocher	f59f80114c	Add 'exclude' tuple to check_requirements() (#2041 )	3 years ago
Glenn Jocher	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	3 years ago
Glenn Jocher	08d3119e09	Add histogram equalization fcn (#2049 )	3 years ago
Glenn Jocher	8fcbe43423	Check im.format during dataset caching (#2042 ) * Check im.format during dataset caching * Update datasets.py	3 years ago
Glenn Jocher	3b7feeafdf	Update general.py check_git_status() fix (#2020 )	3 years ago

1 2 3 4 5 ...

349 Commits (6e8c5b767866ecebe08dc1b673537348394680f3)