yolov5

Commit Graph

Author	SHA1	Message	Date
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	10d56d784e	Assert `--image-weights` not combined with DDP (#3275 )	3 years ago
Glenn Jocher	b7cd1f540d	TensorBoard add_graph() fix (#3236 )	3 years ago
Glenn Jocher	60fe54449d	Update train.py (#3099 )	3 years ago
Glenn Jocher	a833ee2a46	Update check_requirements() exclude list (#2974 )	3 years ago
Glenn Jocher	f7bc685c2c	Implement yaml.safe_load() (#2876 ) * Implement yaml.safe_load() * yaml.safe_dump()	3 years ago
Burhan	c949fc86d1	Detection cropping+saving feature addition for detect.py and PyTorch Hub (#2827 ) * Update detect.py * Update detect.py * Update greetings.yml * Update cropping * cleanup * Update increment_path() * Update common.py * Update detect.py * Update detect.py * Update detect.py * Update common.py * cleanup * Update detect.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	3 years ago
Glenn Jocher	6dd1083bbb	Tensorboard model visualization bug fix (#2758 ) This fix should allow for visualizing YOLOv5 model graphs correctly in Tensorboard by uncommenting line 335 in train.py: ```python if tb_writer: tb_writer.add_graph(torch.jit.trace(model, imgs, strict=False), []) # add model graph ``` The problem was that the detect() layer checks the input size to adapt the grid if required, and tracing does not seem to like this shape check (even if the shape is fine and no grid recomputation is required). The following will warn: `0cae7576a9/train.py (L335)` Solution is below. This is a YOLOv5s model displayed in TensorBoard. You can see the Detect() layer merging the 3 layers into a single output for example, and everything appears to work and visualize correctly. ```python tb_writer.add_graph(torch.jit.trace(model, imgs, strict=False), []) ``` <img width="893" alt="Screenshot 2021-04-11 at 01 10 09" src="https://user-images.githubusercontent.com/26833433/114286928-349bd600-9a63-11eb-941f-7139ee6cd602.png">	3 years ago
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

1 2 3 4 5

226 Commits (0e2f2cbb512fb2120fef7916ab9c2725d3c172f7)