henry000 commited on
Commit
459de68
Β·
1 Parent(s): e9816da

πŸ“ [Update] README with auto remove aux(deploy)

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. yolo/config/model/v9-c-deploy.yaml +0 -77
README.md CHANGED
@@ -61,7 +61,7 @@ python yolo/lazy.py task=train task.data.batch_size=8 model=v9-c dataset={datase
61
  ### Inference
62
  To evaluate the model performance, use:
63
  ```shell
64
- python yolo/lazy.py task=inference weight=weights/v9-c-deploy.pt model=v9-c-deploy # use deploy weight
65
  python python yolo/lazy.py task=inference # if cloned from GitHub
66
  yolo task=inference task.data.source={Any} # if pip installed
67
  ```
 
61
  ### Inference
62
  To evaluate the model performance, use:
63
  ```shell
64
+ python yolo/lazy.py task=inference weight=weights/v9-c.pt model=v9-c task.fast_inference=deploy # use deploy weight
65
  python python yolo/lazy.py task=inference # if cloned from GitHub
66
  yolo task=inference task.data.source={Any} # if pip installed
67
  ```
yolo/config/model/v9-c-deploy.yaml DELETED
@@ -1,77 +0,0 @@
1
- anchor:
2
- reg_max: 16
3
-
4
- model:
5
- backbone:
6
- - Conv:
7
- args: {out_channels: 64, kernel_size: 3, stride: 2}
8
- source: 0
9
- - Conv:
10
- args: {out_channels: 128, kernel_size: 3, stride: 2}
11
- - RepNCSPELAN:
12
- args: {out_channels: 256, part_channels: 128}
13
-
14
- - ADown:
15
- args: {out_channels: 256}
16
- - RepNCSPELAN:
17
- args: {out_channels: 512, part_channels: 256}
18
- tags: B3
19
-
20
- - ADown:
21
- args: {out_channels: 512}
22
- - RepNCSPELAN:
23
- args: {out_channels: 512, part_channels: 512}
24
- tags: B4
25
-
26
- - ADown:
27
- args: {out_channels: 512}
28
- - RepNCSPELAN:
29
- args: {out_channels: 512, part_channels: 512}
30
- tags: B5
31
-
32
- neck:
33
- - SPPELAN:
34
- args: {out_channels: 512}
35
- tags: N3
36
-
37
- - UpSample:
38
- args: {scale_factor: 2, mode: nearest}
39
- - Concat:
40
- source: [-1, B4]
41
- - RepNCSPELAN:
42
- args: {out_channels: 512, part_channels: 512}
43
- tags: N4
44
-
45
- - UpSample:
46
- args: {scale_factor: 2, mode: nearest}
47
- - Concat:
48
- source: [-1, B3]
49
-
50
- head:
51
- - RepNCSPELAN:
52
- args: {out_channels: 256, part_channels: 256}
53
- tags: P3
54
-
55
- - ADown:
56
- args: {out_channels: 256}
57
- - Concat:
58
- source: [-1, N4]
59
- - RepNCSPELAN:
60
- args: {out_channels: 512, part_channels: 512}
61
- tags: P4
62
-
63
- - ADown:
64
- args: {out_channels: 512}
65
- - Concat:
66
- source: [-1, N3]
67
- - RepNCSPELAN:
68
- args: {out_channels: 512, part_channels: 512}
69
- tags: P5
70
-
71
- detection:
72
- - MultiheadDetection:
73
- source: [P3, P4, P5]
74
- tags: Main
75
- args:
76
- reg_max: ${model.anchor.reg_max}
77
- output: True