yolov5

Gráfico de commits

Autor	SHA1	Mensaje	Fecha
Glenn Jocher	60fe54449d	Update train.py (#3099 )	hace 3 años
Glenn Jocher	a833ee2a46	Update check_requirements() exclude list (#2974 )	hace 3 años
Glenn Jocher	f7bc685c2c	Implement yaml.safe_load() (#2876 ) * Implement yaml.safe_load() * yaml.safe_dump()	hace 3 años
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>	hace 3 años
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">	hace 3 años
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>	hace 3 años
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>	hace 3 años
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	hace 3 años
Ayush Chaurasia	dc51e80b00	Fix: evolve with wandb (#2634 )	hace 3 años
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	hace 3 años
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	hace 3 años
Ayush Chaurasia	1bf9365280	W&B DDP fix (#2574 )	hace 3 años
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>	hace 3 años
Glenn Jocher	08d4918d7f	labels.jpg class names (#2454 ) * labels.png class names * fontsize=10	hace 3 años
Glenn Jocher	f01f3223d5	Integer printout (#2450 ) * Integer printout * test.py 'Labels' * Update train.py	hace 3 años
Glenn Jocher	886f1c03d8	DDP after autoanchor reorder (#2421 )	hace 3 años
Glenn Jocher	e931b9da33	Resume with custom anchors fix (#2361 ) * Resume with custom anchors fix * Update train.py	hace 3 años
Glenn Jocher	a3ecf0fd64	Anchor override (#2350 )	hace 3 años
Glenn Jocher	fab5085674	EMA bug fix 2 (#2330 ) * EMA bug fix 2 * update	hace 3 años
Glenn Jocher	dfeec198cb	final_epoch EMA bug fix (#2317 )	hace 3 años
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>	hace 3 años
Glenn Jocher	71dd2768f2	Improved model+EMA checkpointing 2 (#2295 )	hace 3 años
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	hace 3 años
Glenn Jocher	ca5b10b759	Update train.py (#2290 ) * Update train.py * Update train.py * Update train.py * Update train.py * Create train.py	hace 3 años
Glenn Jocher	e27ca0d845	Update minimum stride to 32 (#2266 )	hace 3 años
Glenn Jocher	6b634c6b87	Linear LR scheduler option (#2150 ) * Linear LR scheduler option * Update train.py	hace 3 años
Glenn Jocher	ad839eda38	Update train.py (#2149 )	hace 3 años
Glenn Jocher	86897e3663	Update train.py test batch_size (#2148 ) * Update train.py * Update loss.py	hace 3 años
Glenn Jocher	2acbe9699a	W&B epoch logging update (#2073 )	hace 3 años
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	hace 3 años
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	hace 3 años
Glenn Jocher	046c37e465	verbose on final_epoch (#1997 )	hace 3 años
Abhiram V	e9941d50fa	Update train.py with yaml.SafeLoader (#1972 )	hace 3 años
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>	hace 3 años
Glenn Jocher	ca9babb8e6	Add ComputeLoss() class (#1950 )	hace 3 años
Glenn Jocher	f4a78e1b40	W&B mosaic log bug fix (#1949 )	hace 3 años
Glenn Jocher	777032efbc	Daemon thread mosaic plots fix (#1943 )	hace 3 años
NanoCode012	3a56cac414	Fix batch-size on resume for multi-gpu (#1942 )	hace 3 años
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	hace 3 años
Glenn Jocher	dd03b20ba5	colorstr() updates (#1909 ) * W&B ImportError message fix * colorstr() updates * colorstr() updates * colorstr() default to 'blue', 'bold' * train: magenta * train: blue	hace 3 años
Glenn Jocher	1d1c0567a4	PyTorch Hub results.render() (#1897 )	hace 3 años
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()	hace 3 años
Glenn Jocher	e77c77f580	Add check_requirements() (#1853 ) * Add check_requirements() * add import * parameterize filename * add to detect, test	hace 3 años
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	hace 3 años
Glenn Jocher	0e341c5660	Create one_cycle() function (#1836 )	hace 3 años
Glenn Jocher	7180b22e2d	DDP Multi-GPU --resume bug fix (#1810 )	hace 3 años
Glenn Jocher	685d601308	Increase plot_labels() speed (#1736 )	hace 3 años
Glenn Jocher	7e161d9774	Single class train update (#1719 )	hace 3 años
Glenn Jocher	c923fbff90	W&B artifacts feature addition (#1712 ) * Log artifacts * cleanup	hace 3 años
Glenn Jocher	8bc0027afc	Update loss criteria constructor (#1711 )	hace 3 años

1 2 3 4 5

222 Commits (06372b1465f5a58463bf8c32bdf65fc679c17ebf)