d2da523053
Merge remote-tracking branch 'origin/master'
2020-08-11 20:11:58 -07:00
9f482cbcb8
utils.general comment updates/bug fixes
2020-08-11 20:11:44 -07:00
1f92422e20
Github release assets model autodownload ( #711 )
...
* assets autodownload
* Update google_utils.py
* Update google_utils.py
* Update google_utils.py
* Update google_utils.py
* Update google_utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-11 17:17:24 -07:00
4949401a94
Fix redundant outputs via Logging in DDP training ( #500 )
...
* Change print to logging
* Clean function set_logging
* Add line spacing
* Change leftover prints to log
* Fix scanning labels output
* Fix rank naming
* Change leftover print to logging
* Reorganized DDP variables
* Fix type error
* Make quotes consistent
* Fix spelling
* Clean function call
* Add line spacing
* Update datasets.py
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-11 11:18:45 -07:00
542833c997
Filter the None labels ( #705 )
...
* the value should be a list to unpacking
* filter the None labels
2020-08-11 11:15:48 -07:00
9ae868364a
the value should be a list to unpacking ( #688 )
2020-08-10 11:23:48 -07:00
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
2020-08-09 20:52:57 -07:00
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
2020-08-09 02:12:44 -07:00
3fcd365ba3
Fix curl download on Windows ( #669 )
2020-08-07 12:13:55 -07:00
fe5b3f8712
Fix curl download on Windows ( #669 )
...
* Update ci-testing.yml
* Fix windows download
* Fix cookie and curl download issue from gdrive
* Revert "Update ci-testing.yml"
This reverts commit 7389d2238d
2020-08-07 12:10:59 -07:00
7eaf225d55
zero-target training bug fix ( #609 )
2020-08-05 13:35:31 -07:00
f346da9f2b
update activations.py
2020-08-03 17:18:31 -07:00
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
2020-08-02 15:47:36 -07:00
ec7a926163
AutoAnchor update to display anchors/target
2020-08-02 14:38:45 -07:00
d989bc9260
remove NBSP
2020-08-02 14:23:05 -07:00
023e37807c
hyperparameter evolution update ( #566 )
2020-08-02 13:18:03 -07:00
64e6d199ba
hyperparameter evolution update ( #566 )
2020-08-02 13:17:26 -07:00
305c6a028a
compute_loss() leaf variable update
2020-08-02 10:30:43 -07:00
c1a2a7a411
hyperparameter evolution bug fix ( #566 )
2020-08-01 23:00:10 -07:00
e32abb5fb9
hyperparameter evolution bug fix ( #566 )
2020-08-01 19:15:48 -07:00
127cbeb3f5
hyperparameter expansion to flips, perspective, mixup
2020-08-01 13:47:54 -07:00
23d2bde01f
leaf variable --single-cls training bug fix ( #593 )
2020-08-01 11:42:41 -07:00
bcd452c482
replace random_affine() with random_perspective()
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-31 15:53:52 -07:00
bb87276d80
update build_targets() ( #589 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-31 14:34:13 -07:00
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
2020-07-31 10:52:45 -07:00
c8892672a7
Funnel ReLU (FReLU) @staticmethod bug fix ( #556 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 17:38:15 -07:00
131782a2aa
Funnel ReLU (FReLU) ( #556 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 17:35:34 -07:00
4b074d9d9d
Funnel ReLU (FReLU) ( #556 )
...
* fix #541 #542
* Update train.py
* Add Frelu
* Update activations.py
PEP8 and format updates for commonality with models.common.Conv()
* Update activations.py
Update case
* Update activations.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 17:25:49 -07:00
7d78b17a69
Fix utils.py nonzero() warning ( #551 )
2020-07-29 12:14:11 -07:00
a209a32019
Fix bug #541 #542 ( #545 )
...
* fix #541 #542
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-28 18:31:01 -07:00
0032af2980
Update utils.py strip_optimizer() ( #509 )
2020-07-24 19:26:56 -07:00
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
...
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
2020-07-24 11:42:23 -07:00
4b5f4806bc
Update datasets.py ( #494 )
2020-07-23 22:49:54 -07:00
9da56b62dd
v2.0 Release ( #491 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-23 15:34:23 -07:00
4ffd9779d3
plotting improvements ( #471 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-21 14:07:11 -07:00
1e95337f3a
Update datasets.py ( #454 )
2020-07-20 10:33:09 -07:00
07a82f4d44
Update ci-testing.yml ( #445 )
...
* Update ci-testing.yml
* Update ci-testing.yml
* Update requirements.txt
* Update requirements.txt
* Update google_utils.py
* Update test.py
* Update ci-testing.yml
2020-07-19 15:44:11 -07:00
4102fcc9a7
[WIP] Feature/ddp fixed ( #401 )
...
* Squashed commit of the following:
commit d738487089e41c22b3b1cd73aa7c1c40320a6ebf
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 e742dd9619d29306c7541821238d3d7cddcdc508
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 15:38:48 2020 +0800
Make SyncBN a choice
commit e90d4004387e6103fecad745f8cbc2edc918e906
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 cd9036017e7f8bd519a8b62adab0f47ea67f4962
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 13:39:29 2020 +0700
Update train.py
Remove redundant `opt.` prefix.
commit 5bf8bebe8873afb18b762fe1f409aca116fac073
Merge: c9558a9 a1c8406https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit c9558a9b51547febb03d9c1ca42e2ef0fc15bb31
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 13:51:34 2020 +0800
Add device allocation for loss compute
commit 4f08c692fb5e943a89e0ee354ef6c80a50eeb28d
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:16:27 2020 +0800
Revert drop_last
commit 1dabe33a5a223b758cc761fc8741c6224205a34b
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 a1ce9b1e96b71d7fcb9d3e8143013eb8cebe5e27
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:21 2020 +0800
fix lr warning
commit 4b8450b46db76e5e58cd95df965d4736077cfb0e
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 02c63ef81cf98b28b10344fe2cce08a03b143941
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Wed Jul 8 10:08:30 2020 +0700
Add drop_last for multi gpu
commit b9a50aed48ab1536f94d49269977e2accd67748f
Merge: ec2dc6c 121d90bhttps://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit ec2dc6cc56de43ddff939e14c450672d0fbf9b3d
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 d0326e398dfeeeac611ccc64198d4fe91b7aa969
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:31:24 2020 +0800
Add SyncBN
commit 82a6182b3ad0689a4432b631b438004e5acb3b74
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 050b2a5a79a89c9405854d439a1f70f892139b1c
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:38:14 2020 +0700
Add cleanup for process_group
commit 2aa330139f3cc1237aeb3132245ed7e5d6da1683
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 77c8e27e603bea9a69e7647587ca8d509dc1990d
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 01:54:39 2020 +0700
Convert BatchNorm to SyncBatchNorm
commit 96fa40a3a925e4ffd815fe329e1b5181ec92adc8
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 21:53:56 2020 +0800
Fix the datset inconsistency problem
commit 16e7c269d062c8d16c4d4ff70cc80fd87935dc95
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 e83805563065ffd2e38f85abe008fc662cc17909
Merge: 625bb49 3bdea3fhttps://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit 625bb49f4e52d781143fea0af36d14e5be8b040c
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 2 22:45:15 2020 +0800
DDP established
* Squashed commit of the following:
commit 94147314e559a6bdd13cb9de62490d385c27596f
Merge: 65157e2 37acbdchttps://github.com/ultralytics/yolov4 into feature/DDP_fixed
commit 37acbdc0b6b8c2da4a0dhttps://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit cd55b445c4dcd8003ff4b0b46b64adf7c16e5ce7
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Wed Jul 15 16:42:33 2020 +0800
fix the DDP performance deterioration bug.
commit 0f3b8bb1fae5885474ba861bbbd1924fb622ee93
Author: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed Jul 15 00:28:53 2020 -0700
Delete README.md
commit f5921ba1e35475f24b062456a890238cb7a3cf94
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 bd3fdbbf1b08ef87931eef49fa8340621caa7e87
Author: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue Jul 14 18:38:20 2020 -0700
Update README.md
commit c1a97a7767ccb2aa9afc7a5e72fd159e7c62ec02
Merge: 2bf86b8 f796708https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit 5a19011949398d06e744d8d5521ab4e6dfa06ab7
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 22:19:15 2020 +0800
Add assertion for <=2 gpus DDP
commit c8357ad5b15a0e6aeef4d7fe67ca9637f7322a4d
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 787582f97251834f955ef05a77072b8c673a8397
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 20:38:58 2020 +0700
Fixed issue with single gpu not having world_size
commit 63648925288d63a21174a4dd28f92dbfebfeb75a
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 69364d6050e048d0d8834e0f30ce84da3f6a13f3
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 17:36:48 2020 +0700
Changed number of workers check
commit d738487089e41c22b3b1cd73aa7c1c40320a6ebf
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 e742dd9619d29306c7541821238d3d7cddcdc508
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 15:38:48 2020 +0800
Make SyncBN a choice
commit e90d4004387e6103fecad745f8cbc2edc918e906
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 cd9036017e7f8bd519a8b62adab0f47ea67f4962
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 13:39:29 2020 +0700
Update train.py
Remove redundant `opt.` prefix.
commit 5bf8bebe8873afb18b762fe1f409aca116fac073
Merge: c9558a9 a1c8406https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit c9558a9b51547febb03d9c1ca42e2ef0fc15bb31
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 13:51:34 2020 +0800
Add device allocation for loss compute
commit 4f08c692fb5e943a89e0ee354ef6c80a50eeb28d
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:16:27 2020 +0800
Revert drop_last
commit 1dabe33a5a223b758cc761fc8741c6224205a34b
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 a1ce9b1e96b71d7fcb9d3e8143013eb8cebe5e27
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:21 2020 +0800
fix lr warning
commit 4b8450b46db76e5e58cd95df965d4736077cfb0e
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 02c63ef81cf98b28b10344fe2cce08a03b143941
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Wed Jul 8 10:08:30 2020 +0700
Add drop_last for multi gpu
commit b9a50aed48ab1536f94d49269977e2accd67748f
Merge: ec2dc6c 121d90bhttps://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit ec2dc6cc56de43ddff939e14c450672d0fbf9b3d
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 d0326e398dfeeeac611ccc64198d4fe91b7aa969
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:31:24 2020 +0800
Add SyncBN
commit 82a6182b3ad0689a4432b631b438004e5acb3b74
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 050b2a5a79a89c9405854d439a1f70f892139b1c
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:38:14 2020 +0700
Add cleanup for process_group
commit 2aa330139f3cc1237aeb3132245ed7e5d6da1683
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 77c8e27e603bea9a69e7647587ca8d509dc1990d
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 01:54:39 2020 +0700
Convert BatchNorm to SyncBatchNorm
commit 96fa40a3a925e4ffd815fe329e1b5181ec92adc8
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 21:53:56 2020 +0800
Fix the datset inconsistency problem
commit 16e7c269d062c8d16c4d4ff70cc80fd87935dc95
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 e83805563065ffd2e38f85abe008fc662cc17909
Merge: 625bb49 3bdea3fhttps://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit 625bb49f4e52d781143fea0af36d14e5be8b040c
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>
2020-07-19 12:33:30 -07:00
aa08b2bd6f
test.py --task test updates
2020-07-15 18:25:34 -07:00
98f4a5af87
runs/exp increment bug fix #344
2020-07-15 16:07:40 -07:00
2377e5f6ee
FP16 EMA bug fix
2020-07-13 15:47:46 -07:00
140d84cca1
comment updates
2020-07-13 12:17:52 -07:00
8412d44dce
update datasets.py to add detect.py --source glob_command feature
2020-07-12 14:22:15 -07:00
eae33303d6
update datasets.py LoadImages() path improvements and Mixup
2020-07-12 14:14:51 -07:00
41ff83a98d
update torch_utils.py to FP16 EMA
2020-07-12 12:18:00 -07:00
f7a6d6c8d4
remove https:// prefix from curl commands fix #355
2020-07-11 13:00:24 -07:00
a586751904
multi-gpu ckpt filesize bug fix #253
2020-07-11 12:35:21 -07:00
d6f6b33a49
update datasets.py for image size check during caching
2020-07-11 00:21:52 -07:00
01a67a9742
Merge remote-tracking branch 'origin/master'
2020-07-10 20:36:26 -07:00
d3e786ed93
google_utils.py bug fix #355
2020-07-10 20:36:10 -07:00
41bb70b325
update create_pretrained()
2020-07-10 00:25:35 -07:00
1531795f5e
avoid git status check in Docker bug fix #342
2020-07-10 00:11:50 -07:00
1e9cf6a6e9
add file sizes to ckpt ops
2020-07-10 00:00:19 -07:00
59beae98e8
add if isfile() to get_hash()
2020-07-09 20:57:24 -07:00
6649770001
--rect test bug fix
2020-07-09 20:39:11 -07:00
520f5de6f0
Label caching foundational re-write #306
2020-07-09 20:07:16 -07:00
dd8e742ece
Merge pull request #334 from lorenzomammana/feature-multiple-datasets-training
...
Handle multiple datasets
2020-07-09 16:29:21 -07:00
dd33d2ab77
Update datasets.py
2020-07-09 16:28:20 -07:00
2110f5ee89
update log_dir to runs/exp #107
2020-07-09 16:03:10 -07:00
603ea0bfdc
update log_dir to runs/exp #107
2020-07-09 15:58:07 -07:00
24c5a941f0
--resume EMA fix #292
2020-07-09 15:09:06 -07:00
13a09fa68a
Update datasets.py
2020-07-09 13:47:20 -07:00
47bf1730fa
Update datasets.py
2020-07-09 13:45:55 -07:00
f9bab6b127
Update datasets.py
2020-07-09 13:36:23 -07:00
54a9e4f876
Refactor code to reduce duplication
2020-07-09 19:39:28 +02:00
94342acbe1
Handle multiple datasets
2020-07-09 11:52:12 +02:00
bf6f41567a
hyperparameter printout update
2020-07-08 17:21:00 -07:00
dc5e18390a
Merge branch 'master' into advanced_logging
2020-07-08 17:01:19 -07:00
cbe39a1dbb
Update utils.py
2020-07-08 16:14:32 -07:00
c654d18f00
Update utils.py
2020-07-08 16:11:22 -07:00
31b07b3550
Merge pull request #275 from wanghaoyang0106/master
...
[bug fix] potential problem if img fed to model is in rectangular shape
2020-07-08 00:07:54 -07:00
e8cf24b6c8
Initial model ensemble capability #318
2020-07-07 15:40:50 -07:00
95f0a56df7
Bug fix to get_latest_run when recent run is named
2020-07-07 11:19:49 -05:00
f517ba81c7
Merge branch 'master' into advanced_logging
2020-07-06 16:52:11 -05:00
121d90b3f2
update fuse_conv_and_bn()
2020-07-06 11:46:10 -07:00
38f5c1ad1d
pruning and sparsity initial commit
2020-07-05 13:41:21 -07:00
ccfa9373ae
Update to remove "module" attribute check
2020-07-04 23:12:57 +07:00
f83521ab9a
Fix save error for multi-gpu
2020-07-04 16:50:33 +07:00
ce0c58f678
update compute_loss()
2020-07-03 18:56:07 -07:00
af41083d52
EMA FP16 fix #279
2020-07-03 16:57:08 -07:00
f767023c56
offset and balance update
2020-07-03 14:07:00 -07:00
df224a0d8f
EMA bug fix #279
2020-07-03 11:56:14 -07:00
b57f83d005
[bug fix] potential problem if img fed to model is in rectangular shape
2020-07-03 13:09:21 +08:00
f02481c73a
Update torch_utils.py
2020-07-02 12:03:45 -07:00
fc7c42723d
Update utils.py
2020-07-02 12:01:43 -07:00
53cdaf6bf5
Update utils.py
2020-07-02 13:48:19 +08:00
9a9333d245
add replicate() to datasets.py
2020-06-29 17:10:33 -07:00
37e13f8846
update mosaic border
2020-06-27 13:50:15 -07:00
93a6765806
update mosaic border
2020-06-27 13:02:01 -07:00
e18e6811dc
Merge branch 'master' into advanced_logging
2020-06-27 10:13:03 -05:00
77fb8ee082
refactor dataloader
2020-06-26 19:30:09 -07:00
22fb2b0c25
refactor dataloader
2020-06-26 18:56:13 -07:00
b8557f87e3
add stride to datasets.py
2020-06-24 13:02:27 -07:00
3b062254a6
update utils.create_pretrained()
2020-06-24 12:53:43 -07:00
6c1b87a42e
update google_utils import
2020-06-22 23:00:23 -07:00
364fcfd7db
PANet update
2020-06-22 14:27:17 -07:00
fc171e2678
check_anchor_order() update
2020-06-21 14:08:59 -07:00
1f1917ef56
remove fast, add merge
2020-06-21 13:37:11 -07:00
e572bb0803
Add plot_results save location to log_dir
2020-06-21 09:36:28 -05:00
024a42f4fd
datasets.py vid_formats updated
2020-06-20 09:53:37 -07:00
5c2cd711df
vid_formats bug fix #146
2020-06-20 09:26:07 -07:00
10145833c3
strip_optimizer() default to save FP16 model
2020-06-18 00:16:47 -07:00
cce95e744d
backbone as FP16, save default to FP32
2020-06-18 00:13:18 -07:00
9fdb0fbacf
AutoAnchor bug fix # 117
2020-06-17 19:51:15 -07:00
945307beba
Add save_dir to plot_lr_scheduler and plot_labels
...
Set save_dir = log_dir in train.py
2020-06-17 16:03:18 -05:00
3263a204ea
Fix get_latest_run to search 'search_dir' recursivly
2020-06-17 10:34:37 -05:00
d34291733b
Fix get_latest_run() to search inside 'weights' subfolders
2020-06-17 10:26:55 -05:00
a557b7d924
check_img_size() update
2020-06-16 17:48:01 -07:00
034609414e
AutoAnchor fitness to screen
2020-06-16 17:08:45 -07:00
ef75f7313d
AutoAnchor n, imgsz bug fix
2020-06-16 16:48:01 -07:00
1dfc28527f
AutoAnchor BPR to 4 significant figures
2020-06-16 16:19:58 -07:00
4052603e1f
AutoAnchor update - improvement check
2020-06-16 15:54:20 -07:00
bafbc65ee3
AutoAnchor bug fix
2020-06-16 14:26:02 -07:00
25e51bcec7
add util function to get most recent last.pt file
...
added logic in train.py __main__ to handle resuming from a run
2020-06-16 15:50:27 -05:00
490f1e7b9c
add save_dir arg to plot_lr_scheduler, default to current dir.
...
Uncomment plot_lr_scheduler in train() and pass log_dir as save location
2020-06-16 15:13:03 -05:00
57a0ae3350
AutoAnchor implementation
2020-06-16 13:12:22 -07:00
95c46f7245
Merge pull request #79 from Lornatang/delete-redundant-thrid-party-library
...
Delete redundant thrid party library
2020-06-16 11:52:36 -07:00
ec81c7b5f2
check_anchors() bug fix #90
2020-06-16 10:14:04 -07:00
8b26e89006
AutoAnchor bug fix #72
2020-06-16 00:53:34 -07:00
8fa3724072
kmean_anchors() update
2020-06-16 00:40:21 -07:00
1c0b6236e3
update fast mode
2020-06-16 00:11:29 -07:00
8b8b792802
There is no need to download extra packages, official bring it with you
...
I submitted it once in your yolov3 project, you seem to accept it? I'm not sure. I'll submit PR again.
2020-06-16 08:28:52 +08:00
db2c3acd3a
updated testing settings, rebalanced towards FP16 latency
2020-06-15 16:55:57 -07:00
bd3e389130
update google_utils.py
2020-06-15 14:39:04 -07:00
c5966abba8
glob search bug fix #77
2020-06-15 12:08:57 -07:00
54795a4f0a
fix problem
2020-06-15 16:11:06 +08:00
260b1729f0
FP16 inference update
2020-06-14 18:32:29 -07:00
93bbf1b87a
best possible recall fix
2020-06-13 15:25:53 -07:00
31f3310029
assert best possible recall > 0.9 before training
2020-06-13 15:05:41 -07:00
099e6f5ebd
--img-size stride-multiple verification
2020-06-12 22:10:46 -07:00
3a5c5328c5
ONNX, BCEBlurWithLogitsLoss, plot_study updates
2020-06-11 14:44:09 -07:00
ac93e9c811
Fix Google Drive URI for YOLOv5 weights
2020-06-10 14:26:10 -07:00
c14368d768
new CSP model release
2020-06-09 14:31:04 -07:00
ff02ae0869
module updates
2020-06-08 22:13:01 -07:00
1f698af6f1
create_backbone with SourceChangeWarning update
2020-06-07 15:45:05 -07:00
8fe299f179
model fuse
2020-06-07 14:10:30 -07:00
b810b21299
augmented inference
2020-06-05 21:14:15 -07:00
c12023f145
updates
2020-06-05 20:50:43 -07:00
66d73e4cf3
NMS updates
2020-06-05 18:44:39 -07:00
948bcdd219
--classes bug fix #17
2020-06-05 13:29:26 -07:00
22d6088205
speed-reproducibility fix #17
2020-06-05 13:07:09 -07:00
0362f103ef
updates
2020-06-04 10:19:06 -07:00
2a0aff685a
dataset padding mode
2020-06-03 13:55:33 -07:00
eb97b2e413
NMS fast mode
2020-06-03 13:02:59 -07:00
ee8988b8a2
datasets.py self.batch_shapes fix
2020-06-01 18:11:18 -07:00
f96e8f3edf
updates
2020-05-31 19:51:32 -07:00
b10609fe6b
updates
2020-05-31 15:07:04 -07:00
72eed215ff
updates
2020-05-31 09:25:42 -07:00
5a79b5f65c
updates
2020-05-30 15:22:09 -07:00
4f44aaf26b
updates
2020-05-30 15:18:59 -07:00
1e84a23f38
initial commit
2020-05-29 17:04:54 -07:00
Glenn Jocher
Marc
NanoCode012
tkianai
Jirka Borovec
Liu Changyu
AlexWang1900
YuriRibeiro
pritul dave
yzchen
lorenzomammana
Alex Stoken
wanghaoyang0106
yxNONG
Lornatang
Trevor Sullivan