yolov5

Commit Graph

Author	SHA1	Message	Date
NanoCode012	7f8471eaeb	--notest bug fix (#518 ) * Fix missing results_file and fi when notest passed * Update train.py reverting previous changes and removing functionality from 'if not opt.notest or final_epoch: # Calculate mAP' loop. Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	4 years ago
Glenn Jocher	9da56b62dd	v2.0 Release (#491 ) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>	4 years ago
Glenn Jocher	5e970d45c4	Update train.py (#462 )	4 years ago
Glenn Jocher	3edc38f603	update train.py gsutil bucket fix (#463 )	4 years ago
Glenn Jocher	776555771f	update train.py gsutil bucket fix (#463 )	4 years ago
Glenn Jocher	b569ed6d6b	pretrained model loading bug fix (#450 ) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>	4 years ago
yzchen	4102fcc9a7	[WIP] Feature/ddp fixed (#401 ) * Squashed commit of the following: commit `d738487089` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 14 17:33:38 2020 +0700 Adding world_size Reduce calls to torch.distributed. For use in create_dataloader. commit `e742dd9619` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 14 15:38:48 2020 +0800 Make SyncBN a choice commit `e90d400438` Merge: 5bf8beb cd90360 Author: yzchen <Chenyzsjtu@gmail.com> Date: Tue Jul 14 15:32:10 2020 +0800 Merge pull request #6 from NanoCode012/patch-5 Update train.py commit `cd9036017e` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 14 13:39:29 2020 +0700 Update train.py Remove redundant `opt.` prefix. commit `5bf8bebe88` Merge: c9558a9 a1c8406 Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 14 14:09:51 2020 +0800 Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed commit `c9558a9b51` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 14 13:51:34 2020 +0800 Add device allocation for loss compute commit `4f08c692fb` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Thu Jul 9 11:16:27 2020 +0800 Revert drop_last commit `1dabe33a5a` Merge: a1ce9b1 4b8450b Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Thu Jul 9 11:15:49 2020 +0800 Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed commit `a1ce9b1e96` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Thu Jul 9 11:15:21 2020 +0800 fix lr warning commit `4b8450b46d` Merge: b9a50ae 02c63ef Author: yzchen <Chenyzsjtu@gmail.com> Date: Wed Jul 8 21:24:24 2020 +0800 Merge pull request #4 from NanoCode012/patch-4 Add drop_last for multi gpu commit `02c63ef81c` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Wed Jul 8 10:08:30 2020 +0700 Add drop_last for multi gpu commit `b9a50aed48` Merge: ec2dc6c 121d90b Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 7 19:48:04 2020 +0800 Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed commit `ec2dc6cc56` Merge: d0326e3 82a6182 Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 7 19:34:31 2020 +0800 Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed commit `d0326e398d` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 7 19:31:24 2020 +0800 Add SyncBN commit `82a6182b3a` Merge: 96fa40a 050b2a5 Author: yzchen <Chenyzsjtu@gmail.com> Date: Tue Jul 7 19:21:01 2020 +0800 Merge pull request #1 from NanoCode012/patch-2 Convert BatchNorm to SyncBatchNorm commit `050b2a5a79` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 7 12:38:14 2020 +0700 Add cleanup for process_group commit `2aa330139f` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 7 12:07:40 2020 +0700 Remove apex.parallel. Use torch.nn.parallel For future compatibility commit `77c8e27e60` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 7 01:54:39 2020 +0700 Convert BatchNorm to SyncBatchNorm commit `96fa40a3a9` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Mon Jul 6 21:53:56 2020 +0800 Fix the datset inconsistency problem commit `16e7c269d0` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Mon Jul 6 11:34:03 2020 +0800 Add loss multiplication to preserver the single-process performance commit `e838055630` Merge: 625bb49 3bdea3f Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Fri Jul 3 20:56:30 2020 +0800 Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed commit `625bb49f4e` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Thu Jul 2 22:45:15 2020 +0800 DDP established * Squashed commit of the following: commit `94147314e5` Merge: 65157e2 37acbdc Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Thu Jul 16 14:00:17 2020 +0800 Merge branch 'master' of https://github.com/ultralytics/yolov4 into feature/DDP_fixed commit `37acbdc0b6` Author: Glenn Jocher <glenn.jocher@ultralytics.com> Date: Wed Jul 15 20:03:41 2020 -0700 update test.py --save-txt commit `b8c2da4a0d` Author: Glenn Jocher <glenn.jocher@ultralytics.com> Date: Wed Jul 15 20:00:48 2020 -0700 update test.py --save-txt commit `65157e2fc9` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Wed Jul 15 16:44:13 2020 +0800 Revert the README.md removal commit `1c802bfa50` Merge: cd55b44 0f3b8bb Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Wed Jul 15 16:43:38 2020 +0800 Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed commit `cd55b445c4` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Wed Jul 15 16:42:33 2020 +0800 fix the DDP performance deterioration bug. commit `0f3b8bb1fa` Author: Glenn Jocher <glenn.jocher@ultralytics.com> Date: Wed Jul 15 00:28:53 2020 -0700 Delete README.md commit `f5921ba1e3` Merge: 85ab2f3 bd3fdbb Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Wed Jul 15 11:20:17 2020 +0800 Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed commit `bd3fdbbf1b` Author: Glenn Jocher <glenn.jocher@ultralytics.com> Date: Tue Jul 14 18:38:20 2020 -0700 Update README.md commit `c1a97a7767` Merge: 2bf86b8 f796708 Author: Glenn Jocher <glenn.jocher@ultralytics.com> Date: Tue Jul 14 18:36:53 2020 -0700 Merge branch 'master' into feature/DDP_fixed commit 2bf86b892fa2fd712f6530903a0d9b8533d7447a Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 14 22:18:15 2020 +0700 Fixed world_size not found when called from test commit 85ab2f38cdda28b61ad15a3a5a14c3aafb620dc8 Merge: 5a19011 c8357ad Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 14 22:19:58 2020 +0800 Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed commit `5a19011949` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 14 22:19:15 2020 +0800 Add assertion for <=2 gpus DDP commit `c8357ad5b1` Merge: e742dd9 787582f Author: yzchen <Chenyzsjtu@gmail.com> Date: Tue Jul 14 22:10:02 2020 +0800 Merge pull request #8 from MagicFrogSJTU/NanoCode012-patch-1 Modify number of dataloaders' workers commit `787582f972` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 14 20:38:58 2020 +0700 Fixed issue with single gpu not having world_size commit `6364892528` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 14 19:16:15 2020 +0700 Add assert message for clarification Clarify why assertion was thrown to users commit `69364d6050` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 14 17:36:48 2020 +0700 Changed number of workers check commit `d738487089` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 14 17:33:38 2020 +0700 Adding world_size Reduce calls to torch.distributed. For use in create_dataloader. commit `e742dd9619` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 14 15:38:48 2020 +0800 Make SyncBN a choice commit `e90d400438` Merge: 5bf8beb cd90360 Author: yzchen <Chenyzsjtu@gmail.com> Date: Tue Jul 14 15:32:10 2020 +0800 Merge pull request #6 from NanoCode012/patch-5 Update train.py commit `cd9036017e` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 14 13:39:29 2020 +0700 Update train.py Remove redundant `opt.` prefix. commit `5bf8bebe88` Merge: c9558a9 a1c8406 Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 14 14:09:51 2020 +0800 Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed commit `c9558a9b51` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 14 13:51:34 2020 +0800 Add device allocation for loss compute commit `4f08c692fb` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Thu Jul 9 11:16:27 2020 +0800 Revert drop_last commit `1dabe33a5a` Merge: a1ce9b1 4b8450b Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Thu Jul 9 11:15:49 2020 +0800 Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed commit `a1ce9b1e96` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Thu Jul 9 11:15:21 2020 +0800 fix lr warning commit `4b8450b46d` Merge: b9a50ae 02c63ef Author: yzchen <Chenyzsjtu@gmail.com> Date: Wed Jul 8 21:24:24 2020 +0800 Merge pull request #4 from NanoCode012/patch-4 Add drop_last for multi gpu commit `02c63ef81c` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Wed Jul 8 10:08:30 2020 +0700 Add drop_last for multi gpu commit `b9a50aed48` Merge: ec2dc6c 121d90b Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 7 19:48:04 2020 +0800 Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed commit `ec2dc6cc56` Merge: d0326e3 82a6182 Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 7 19:34:31 2020 +0800 Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed commit `d0326e398d` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Tue Jul 7 19:31:24 2020 +0800 Add SyncBN commit `82a6182b3a` Merge: 96fa40a 050b2a5 Author: yzchen <Chenyzsjtu@gmail.com> Date: Tue Jul 7 19:21:01 2020 +0800 Merge pull request #1 from NanoCode012/patch-2 Convert BatchNorm to SyncBatchNorm commit `050b2a5a79` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 7 12:38:14 2020 +0700 Add cleanup for process_group commit `2aa330139f` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 7 12:07:40 2020 +0700 Remove apex.parallel. Use torch.nn.parallel For future compatibility commit `77c8e27e60` Author: NanoCode012 <kevinvong@rocketmail.com> Date: Tue Jul 7 01:54:39 2020 +0700 Convert BatchNorm to SyncBatchNorm commit `96fa40a3a9` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Mon Jul 6 21:53:56 2020 +0800 Fix the datset inconsistency problem commit `16e7c269d0` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Mon Jul 6 11:34:03 2020 +0800 Add loss multiplication to preserver the single-process performance commit `e838055630` Merge: 625bb49 3bdea3f Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Fri Jul 3 20:56:30 2020 +0800 Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed commit `625bb49f4e` Author: yizhi.chen <chenyzsjtu@outlook.com> Date: Thu Jul 2 22:45:15 2020 +0800 DDP established * Fixed destroy_process_group in DP mode * Update torch_utils.py * Update utils.py Revert build_targets() to current master. * Update datasets.py * Fixed world_size attribute not found Co-authored-by: NanoCode012 <kevinvong@rocketmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	4 years ago
Glenn Jocher	07493a715c	update train.py class count assertion #424 Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>	4 years ago
Glenn Jocher	912c06a67c	update train.py class count assertion #424	4 years ago
Glenn Jocher	65857adf52	update train.py ckpt loading	4 years ago
Jirka	e8ea772384	revert test module to confuse users...	4 years ago
Jirka	ef6f5b33a8	rename test	4 years ago
Jirka	9a880a4554	rename eval	4 years ago
Jirka	df270ea9d7	rename eval	4 years ago
Glenn Jocher	03489aaafb	comment tb_writer.add_hparams(hyp, {})	4 years ago
Glenn Jocher	120d40c06a	Update train.py This updates the PR to a one-liner to minimize additions. Perhaps we can include opt in the future but let's start with this for now.	4 years ago
Janko Ondras	38acc5f3c5	Fix img_size naming in hyperparameters logging	4 years ago
Glenn Jocher	a1c8406af3	EMA and non_blocking=True	4 years ago
Glenn Jocher	140d84cca1	comment updates	4 years ago
Janko Ondras	e558963244	Log hyperparameters in tensorboard Log both hyperparameters and command line options in tensorboard.	4 years ago
Glenn Jocher	01a73ec08e	multi-gpu ckpt filesize bug fix #253	4 years ago
Glenn Jocher	a586751904	multi-gpu ckpt filesize bug fix #253	4 years ago
Glenn Jocher	5de4e25d68	update tensorboard metric	4 years ago
Glenn Jocher	98fc483abc	train.py results.txt to bucket bug fix	4 years ago
Glenn Jocher	1c13e67b33	evolution bug fix #346	4 years ago
Laughing	51a161d9e1	Update train.py	4 years ago
Glenn Jocher	e16e9e43e1	new nc=len(names) check	4 years ago
Glenn Jocher	cb527d3af9	new nc=len(names) check	4 years ago
Glenn Jocher	603ea0bfdc	update log_dir to runs/exp #107	4 years ago
Glenn Jocher	72d5b58b9a	disable LR plot to suppress warning message	4 years ago
Glenn Jocher	24c5a941f0	--resume EMA fix #292	4 years ago
Alex Stoken	9d631408a2	Move hyp and opt yaml save to top of train() Fixes bug where scaled values were saved in hyp.yaml, which would cause continuity issues with --resume	4 years ago
Glenn Jocher	bf6f41567a	hyperparameter printout update	4 years ago
Glenn Jocher	6b134d93c5	Update train.py	4 years ago
Glenn Jocher	16f6834486	update train.py and experimental.py	4 years ago
Alex Stoken	52bac22f09	Add in --resume functionality with option to specify path or to get most recent run	4 years ago
Laughing	956511dafd	fix LR bug	4 years ago
Glenn Jocher	bb3c346916	model.yaml nc inherited from dataset.yaml	4 years ago
Glenn Jocher	df224a0d8f	EMA bug fix #279	4 years ago
Glenn Jocher	3bdea3f697	strip_optimizer() bug fix #253	4 years ago
Glenn Jocher	597ed4ce63	Update train.py	4 years ago
Glenn Jocher	9705c15da1	Update train.py	4 years ago
ChristopherSTAN	a8d17ce712	Update train.py	4 years ago
Glenn Jocher	13f69777a6	typo fix	4 years ago
yxNONG	1aa2b67933	Update train.py	4 years ago
Glenn Jocher	86784cfdbf	--resume bug fix #252	4 years ago
Glenn Jocher	ad4c22cbfe	--resume bug fix #187	4 years ago
Glenn Jocher	3b16c865f0	assert --epochs	4 years ago
yxNONG	cdb9bde181	Unify the check point of single and multi GPU save the model.hyp etc to checkpoint when use multi GPU training	4 years ago
Glenn Jocher	b203c9b7ff	update train.py incompatible model message fix #222	4 years ago

1 2 3

133 Commits (a8751e50de94b207c1323c3eda493952cced1d72)