π [Update] README with auto remove aux(deploy)
Browse files- README.md +1 -1
- 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
|
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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|