Skip to content

Commit 65eb404

Browse files
committed
Update README with instructions for adapting model configuration to new class names
1 parent ce75608 commit 65eb404

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,14 @@ You can adjust the log directory using `--logdir <path>` when running `tensorboa
7070
#### Classes
7171
Add class names to `data/custom/yoeo_names.yaml`.
7272

73+
Run the following command to adapt the model file (cfg) to the new number of classes:
74+
75+
```bash
76+
poetry run yoeo-custiomize-cfg -c config/yoeo.cfg -d config/custom.data -o config/yoeo-custom.cfg
77+
```
78+
79+
This changes the layers of the model to fit the number of classes in your dataset.
80+
7381
#### Image Folder
7482
Move the images of your dataset to `data/custom/images/`.
7583

0 commit comments

Comments
 (0)