yolov5

Commit Graph

Author	SHA1	Message	Date
Ding Yiwei	1148e2ea63	Add TransformerLayer, TransformerBlock, C3TR modules (#2333 ) * yolotr * transformer block * Remove bias in Transformer * Remove C3T * Remove a deprecated class * put the 2nd LayerNorm into the 2nd residual block * move example model to models/hub, rename to -transformer * Add module comments and TODOs * Remove LN in Transformer * Add comments for Transformer * Solve the problem of MA with DDP * cleanup * cleanup find_unused_parameters * PEP8 reformat Co-authored-by: DingYiwei <846414640@qq.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	3 years ago
Phat Tran	9c803f2f7e	Add --label-smoothing eps argument to train.py (default 0.0) (#2344 ) * Add label smoothing option * Correct data type * add_log * Remove log * Add log * Update loss.py remove comment (too versbose) Co-authored-by: phattran <phat.tranhoang@cyberlogitec.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	3 years ago
Ayush Chaurasia	518c09578e	W&B resume ddp from run link fix (#2579 ) * W&B resume ddp from run link fix * Native DDP W&B support for training, resuming	3 years ago
Ayush Chaurasia	dc51e80b00	Fix: evolve with wandb (#2634 )	3 years ago
Glenn Jocher	9f98201dd9	W&B DDP fix 2 (#2587 ) Revert unintentional change to test batch sizes caused by PR https://github.com/ultralytics/yolov5/pull/2125	3 years ago
Glenn Jocher	e5b0200cd2	Update tensorboard>=2.4.1 (#2576 ) * Update tensorboard>=2.4.1 Update tensorboard version to attempt to address https://github.com/ultralytics/yolov5/issues/2573 (tensorboard logging fail in Docker image). * cleanup	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	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	886f1c03d8	DDP after autoanchor reorder (#2421 )	3 years ago
Glenn Jocher	e931b9da33	Resume with custom anchors fix (#2361 ) * Resume with custom anchors fix * Update train.py	3 years ago
Glenn Jocher	a3ecf0fd64	Anchor override (#2350 )	3 years ago
Glenn Jocher	fab5085674	EMA bug fix 2 (#2330 ) * EMA bug fix 2 * update	3 years ago
Glenn Jocher	dfeec198cb	final_epoch EMA bug fix (#2317 )	3 years ago
Jan Hajek	efa4946d15	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>	3 years ago
Glenn Jocher	71dd2768f2	Improved model+EMA checkpointing 2 (#2295 )	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	ca5b10b759	Update train.py (#2290 ) * Update train.py * Update train.py * Update train.py * Update train.py * Create train.py	3 years ago
Glenn Jocher	e27ca0d845	Update minimum stride to 32 (#2266 )	3 years ago
Glenn Jocher	6b634c6b87	Linear LR scheduler option (#2150 ) * Linear LR scheduler option * Update train.py	3 years ago
Glenn Jocher	ad839eda38	Update train.py (#2149 )	3 years ago
Glenn Jocher	86897e3663	Update train.py test batch_size (#2148 ) * Update train.py * Update loss.py	3 years ago
Glenn Jocher	2acbe9699a	W&B epoch logging update (#2073 )	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	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	046c37e465	verbose on final_epoch (#1997 )	3 years ago
Abhiram V	e9941d50fa	Update train.py with yaml.SafeLoader (#1972 )	3 years ago
huntr.dev \| the place to protect open source	b5d851d653	Security Fix for Arbitrary Code Execution - huntr.dev (#1962 ) Co-authored-by: Anon-Artist <61599526+Anon-Artist@users.noreply.github.com> Co-authored-by: Jamie Slome <jamie@418sec.com>	3 years ago
Glenn Jocher	ca9babb8e6	Add ComputeLoss() class (#1950 )	3 years ago
Glenn Jocher	f4a78e1b40	W&B mosaic log bug fix (#1949 )	3 years ago
Glenn Jocher	777032efbc	Daemon thread mosaic plots fix (#1943 )	3 years ago
NanoCode012	3a56cac414	Fix batch-size on resume for multi-gpu (#1942 )	3 years ago
Glenn Jocher	509dd51aca	check_git_status() improvements (#1916 ) * check_online() * Update general.py * update check_git_status() * reverse rev-parse order * fetch * improved responsiveness * comment * comment * remove hyp['giou'] compat warning	3 years ago
Glenn Jocher	dd03b20ba5	colorstr() updates (#1909 ) * W&B ImportError message fix * colorstr() updates * colorstr() updates * colorstr() default to 'blue', 'bold' * train: magenta * train: blue	3 years ago
Glenn Jocher	1d1c0567a4	PyTorch Hub results.render() (#1897 )	3 years ago
Glenn Jocher	6ab589583c	Add colorstr() (#1887 ) * Add colorful() * update * newline fix * add git description * --always * update loss scaling * update loss scaling 2 * rename to colorstr()	3 years ago
Glenn Jocher	e77c77f580	Add check_requirements() (#1853 ) * Add check_requirements() * add import * parameterize filename * add to detect, test	3 years ago
Glenn Jocher	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	3 years ago
Glenn Jocher	0e341c5660	Create one_cycle() function (#1836 )	3 years ago
Glenn Jocher	7180b22e2d	DDP Multi-GPU --resume bug fix (#1810 )	3 years ago
Glenn Jocher	685d601308	Increase plot_labels() speed (#1736 )	3 years ago
Glenn Jocher	7e161d9774	Single class train update (#1719 )	3 years ago
Glenn Jocher	c923fbff90	W&B artifacts feature addition (#1712 ) * Log artifacts * cleanup	3 years ago
Glenn Jocher	8bc0027afc	Update loss criteria constructor (#1711 )	3 years ago
Glenn Jocher	7947c86b57	Update COCO train postprocessing (#1702 )	3 years ago
Glenn Jocher	791dadb51c	Pycocotools best.pt after COCO train (#1616 ) * Pycocotools best.pt after COCO train * cleanup	3 years ago
Glenn Jocher	b6ed1104a6	Daemon thread plotting (#1561 ) * Daemon thread plotting * remove process_batch * plot after print	4 years ago
Glenn Jocher	9728e2b8ae	--image_weights bug fix (#1524 )	4 years ago
Glenn Jocher	bde5d9aaaa	Update caching (#1496 )	4 years ago

1 2 3 4 5

220 Commits (2c7c075fb10537fa5c0f8781701f81de99d8906d)