 Glenn Jocher
|
cff9263490
|
f.read().strip() (#1551)
|
vor 3 Jahren |
|
Glenn Jocher
|
9fa7f9f598
|
f.read().strip()
|
vor 3 Jahren |
|
Glenn Jocher
|
12499f1c01
|
--image_weights bug fix (#1524)
|
vor 3 Jahren |
|
Glenn Jocher
|
9728e2b8ae
|
--image_weights bug fix (#1524)
|
vor 3 Jahren |
|
Glenn Jocher
|
e9a0ae6f19
|
Cache bug fix (#1513)
* Caching bug fix #1508
* np.zeros((0,5)) x2
|
vor 3 Jahren |
|
Glenn Jocher
|
2026d4c5eb
|
Update caching (#1496)
|
vor 3 Jahren |
|
Glenn Jocher
|
bde5d9aaaa
|
Update caching (#1496)
|
vor 3 Jahren |
|
Glenn Jocher
|
0822cda781
|
Update caching (#1496)
|
vor 3 Jahren |
|
Glenn Jocher
|
89c7a5b8dc
|
Update caching (#1496)
|
vor 3 Jahren |
 igornishka
|
44f42b1589
|
changed prints to logging in utils/datasets (#1315)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
vor 3 Jahren |
|
Glenn Jocher
|
354109c54c
|
Autosplit (#1488)
|
vor 3 Jahren |
|
Glenn Jocher
|
4798e66fdf
|
Autosplit (#1488)
|
vor 3 Jahren |
 Jacobsolawetz
|
8d2d6d2349
|
Update img2label_paths() (#1414)
|
vor 3 Jahren |
|
Glenn Jocher
|
548a98a391
|
Recursive directories dataset capability (#1408)
* Recursive directories dataset capability
* x.split('.')[-1]
* f += glob.glob(f"{p}{os.sep}**.*", recursive=True)
* f += glob.glob(str(p / '**.*'), recursive=True)
* remove . from image and vid formats
* .txt to txt
* str(p / '**' / '*.*')
|
vor 3 Jahren |
|
Glenn Jocher
|
fe341fa44d
|
Utils reorganization (#1392)
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
|
vor 3 Jahren |
|
Glenn Jocher
|
19e2482458
|
PyTorch version to screen and cleanup (#1325)
* Create flatten_recursive() helper function
* cleanup
* print torch version
|
vor 3 Jahren |
|
Glenn Jocher
|
d3e7778151
|
imap bug fix
|
vor 3 Jahren |
|
Glenn Jocher
|
194f16844e
|
Multi-threaded image caching
|
vor 3 Jahren |
|
Glenn Jocher
|
481d46cffb
|
Improved corruption handling during scan and cache (#999)
|
vor 3 Jahren |
|
Glenn Jocher
|
d61930e017
|
Improved corruption handling during scan and cache (#999)
|
vor 3 Jahren |
 Shiwei Song
|
9291daa6f7
|
fix padding for rectangular inference (#1165)
Co-authored-by: swsong <swsong@stratosphere.mobi>
|
vor 3 Jahren |
|
Glenn Jocher
|
3b57cb5641
|
Simplified inference (#1153)
|
vor 3 Jahren |
 haoyuefan
|
7220cee1d1
|
mosaic4 bottom left image fix (#1021)
fix a bug in load_mosaic
|
vor 4 Jahren |
|
Glenn Jocher
|
f1c63e2784
|
add mosaic and warmup to hyperparameters (#931)
|
vor 4 Jahren |
|
Glenn Jocher
|
806e75f2b1
|
self.label_files bug fix (#958)
|
vor 4 Jahren |
|
Glenn Jocher
|
720645d96a
|
LoadImagesAndLabels() /images/ to /labels/ fix (#928)
|
vor 4 Jahren |
|
Glenn Jocher
|
d49c52eee3
|
_RepeatSampler outside of InfiniteDataLoader
|
vor 4 Jahren |
|
Glenn Jocher
|
9c0efdde9f
|
glob.glob(recursive=True) search (#917)
|
vor 4 Jahren |
|
Glenn Jocher
|
d3f9bf2bb7
|
Update datasets.py
|
vor 4 Jahren |
 NanoCode012
|
1e15aad6f9
|
Add InfiniteDataLoader class (#876)
* Add InfiniteDataLoader
Only initializes at first epoch. Saves time.
* Moved class to a better location
|
vor 4 Jahren |
|
Glenn Jocher
|
82ed33a052
|
generalize webcam detection with opt.source.isnumeric() (yolov3 #1462)
|
vor 4 Jahren |
|
Glenn Jocher
|
f5da528d28
|
reformat code
|
vor 4 Jahren |
|
Glenn Jocher
|
916d4aad9a
|
v3.0 Release (#725)
* initial commit
* remove yolov3-spp from test.py study
* update study --img range
* update mAP
* cleanup and speed updates
* update README plot
|
vor 4 Jahren |
 Marc
|
a925f283a7
|
max workers for dataloader (#722)
|
vor 4 Jahren |
|
NanoCode012
|
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>
|
vor 4 Jahren |
 tkianai
|
542833c997
|
Filter the None labels (#705)
* the value should be a list to unpacking
* filter the None labels
|
vor 4 Jahren |
|
tkianai
|
9ae868364a
|
the value should be a list to unpacking (#688)
|
vor 4 Jahren |
 Jirka Borovec
|
d5b6416c87
|
Explicit Imports (#498)
* expand imports
* optimize
* miss
* fix
|
vor 4 Jahren |
|
Glenn Jocher
|
127cbeb3f5
|
hyperparameter expansion to flips, perspective, mixup
|
vor 4 Jahren |
|
Glenn Jocher
|
bcd452c482
|
replace random_affine() with random_perspective()
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
vor 4 Jahren |
|
Glenn Jocher
|
4b5f4806bc
|
Update datasets.py (#494)
|
vor 4 Jahren |
 pritul dave
|
1e95337f3a
|
Update datasets.py (#454)
|
vor 4 Jahren |
 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>
|
vor 4 Jahren |
|
Glenn Jocher
|
aa08b2bd6f
|
test.py --task test updates
|
vor 4 Jahren |
|
Glenn Jocher
|
8412d44dce
|
update datasets.py to add detect.py --source glob_command feature
|
vor 4 Jahren |
|
Glenn Jocher
|
eae33303d6
|
update datasets.py LoadImages() path improvements and Mixup
|
vor 4 Jahren |
|
Glenn Jocher
|
d6f6b33a49
|
update datasets.py for image size check during caching
|
vor 4 Jahren |
|
Glenn Jocher
|
59beae98e8
|
add if isfile() to get_hash()
|
vor 4 Jahren |
|
Glenn Jocher
|
6649770001
|
--rect test bug fix
|
vor 4 Jahren |
|
Glenn Jocher
|
520f5de6f0
|
Label caching foundational re-write #306
|
vor 4 Jahren |
