henry000 commited on
Commit
d972e5a
Β·
2 Parent(s): 2028dc3 a2c4a3f

πŸ”€ [Merge] branch 'SETUP' into DATASET

Browse files
Files changed (3) hide show
  1. LICENSE +21 -0
  2. README.md +13 -11
  3. utils/dataloader.py +1 -1
LICENSE ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MIT License
2
+
3
+ Copyright (c) 2024 Kin-Yiu, Wong
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
README.md CHANGED
@@ -13,20 +13,22 @@ While the project's structure is still being finalized, we ask that potential co
13
 
14
  If you are interested in contributing, please keep an eye on project updates or contact us directly at [[email protected]](mailto:[email protected]) for more information.
15
 
 
16
 
 
17
 
18
  ## To-Do Lists
19
  - [ ] Project Setup
20
  - [X] requirements
21
- - [ ] LICENSE
22
  - [ ] README
23
- - [ ] pytests
24
  - [ ] setup.py/pip install
25
- - [ ] log format
26
  - [ ] hugging face
27
  - [ ] Data proccess
28
  - [ ] Dataset
29
- - [ ] Download script
30
  - [ ] Auto Download
31
  - [ ] xywh, xxyy, xcyc
32
  - [ ] Dataloder
@@ -35,14 +37,14 @@ If you are interested in contributing, please keep an eye on project updates or
35
  - [ ] load model
36
  - [ ] from yaml
37
  - [ ] from github
38
- - [ ] trainer
39
- - [ ] train_one_iter
40
- - [ ] train_one_epoch
41
- - [ ] DDP, EMA, OTA
 
 
42
  - [ ] Run
43
  - [ ] train
44
  - [ ] test
45
  - [ ] demo
46
- - [ ] Configuration
47
- - [ ] hyperparams: dataclass
48
- - [ ] model cfg: yaml
 
13
 
14
  If you are interested in contributing, please keep an eye on project updates or contact us directly at [[email protected]](mailto:[email protected]) for more information.
15
 
16
+ ## Star History
17
 
18
+ [![Star History Chart](https://api.star-history.com/svg?repos=WongKinYiu/yolov9mit&type=Date)](https://star-history.com/#WongKinYiu/yolov9mit&Date)
19
 
20
  ## To-Do Lists
21
  - [ ] Project Setup
22
  - [X] requirements
23
+ - [x] LICENSE
24
  - [ ] README
25
+ - [x] pytests
26
  - [ ] setup.py/pip install
27
+ - [x] log format
28
  - [ ] hugging face
29
  - [ ] Data proccess
30
  - [ ] Dataset
31
+ - [x] Download script
32
  - [ ] Auto Download
33
  - [ ] xywh, xxyy, xcyc
34
  - [ ] Dataloder
 
37
  - [ ] load model
38
  - [ ] from yaml
39
  - [ ] from github
40
+ - [x] trainer
41
+ - [x] train_one_iter
42
+ - [x] train_one_epoch
43
+ - [ ] DDP
44
+ - [x] EMA, OTA
45
+ - [ ] Loss
46
  - [ ] Run
47
  - [ ] train
48
  - [ ] test
49
  - [ ] demo
50
+ - [x] Configuration
 
 
utils/dataloader.py CHANGED
@@ -50,7 +50,7 @@ class YoloDataset(Dataset):
50
  cache[phase_name] = data
51
 
52
  cache.close()
53
- logger.info("Loaded {} cache", phase_name)
54
  data = cache[phase_name]
55
  return data
56
 
 
50
  cache[phase_name] = data
51
 
52
  cache.close()
53
+ logger.info("πŸ“¦ Loaded {} cache", phase_name)
54
  data = cache[phase_name]
55
  return data
56