yolov5

Commit Graph

Author	SHA1	Message	Date
Glenn Jocher	8e5c66579b	update train.py remove save_json final_epoch	4 years ago
Glenn Jocher	41523e2c91	Dataset autodownload feature addition (#685 ) * initial commit * move download scripts into data/scripts * new check_dataset() function in general.py * move check_dataset() out of with context * Update general.py * DDP update * Update general.py	4 years ago
NanoCode012	3d8ed0a76b	Fix missing model.stride in DP and DDP mode (#683 )	4 years ago
Glenn Jocher	a0ac5adb7b	Single-source training update (#680 )	4 years ago
Glenn Jocher	3c6e2f7668	Single-source training (#680 ) * Single-source training * Extract hyperparameters into seperate files * weight decay scientific notation yaml reader bug fix * remove import glob * intersect_dicts() implementation * 'or' bug fix * .to(device) bug fix	4 years ago
NanoCode012	d7cfbc47ab	Fix unrecognized local rank argument (#676 )	4 years ago
Glenn Jocher	93684531c6	train.py --logdir argparser addition (#660 ) * train.py --logdir argparser addition * train.py --logdir argparser addition	4 years ago
NanoCode012	886b9841c8	Add Multi-Node support for DDP Training (#504 ) * Add support for multi-node DDP * Remove local_rank confusion * Fix spacing	4 years ago
lorenzomammana	728efa6576	Fix missing imports (#627 ) * Fix missing imports * Update detect.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	4 years ago
Glenn Jocher	eb99dff9ef	import random bug fix (#614 )	4 years ago
Jirka Borovec	d5b6416c87	Explicit Imports (#498 ) * expand imports * optimize * miss * fix	4 years ago
Glenn Jocher	f1096b2cf7	hyperparameter evolution update (#566 )	4 years ago
Glenn Jocher	c1a2a7a411	hyperparameter evolution bug fix (#566 )	4 years ago
Glenn Jocher	8074745908	hyperparameter evolution bug fix (#566 )	4 years ago
Glenn Jocher	e32abb5fb9	hyperparameter evolution bug fix (#566 )	4 years ago
Glenn Jocher	8056fe2db8	hyperparameter evolution bug fix (#566 )	4 years ago
Glenn Jocher	127cbeb3f5	hyperparameter expansion to flips, perspective, mixup	4 years ago
Glenn Jocher	bcd452c482	replace random_affine() with random_perspective() Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>	4 years ago
Liu Changyu	c020875b17	PyTorch 1.6.0 update with native AMP (#573 ) * PyTorch have Automatic Mixed Precision (AMP) Training. * Fixed the problem of inconsistent code length indentation * Fixed the problem of inconsistent code length indentation * Mixed precision training is turned on by default	4 years ago
Laughing	4e2b9ecc7e	LR --resume repeat bug fix (#565 )	4 years ago
AlexWang1900	a209a32019	Fix bug #541 #542 (#545 ) * fix #541 #542 * Update train.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	4 years ago
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

1 2 3

124 Commits (9ae868364a2d98bd03cecb8ba8f6310c0d11b482)