更新 .gitignore 文件以排除 .pth 文件，调整训练配置中的类别权重

2024-10-15 20:36:37 +08:00 · 2024-10-15 20:36:37 +08:00 · d5012516b6
commit d5012516b6
parent ff6430db8e
4 changed files with 2946 additions and 5 deletions
--- a/.gitignore
+++ b/.gitignore
@ -1,3 +1,4 @@
 *.jpg
 *.png
+*.pth
 __pycache__
--- a/config/train.yaml
+++ b/config/train.yaml
@ -8,9 +8,9 @@ epochs: 100
 # 每一类的占比权重，如果要让每一类的占比权重相同，为1.0即可
 weight:
  - 1.0
-  - 1.9
+  - 2.9
  - 1.0
-  - 1.2
+  - 1.8

 # 数据集存放位置
 root:
--- a/pretrained/holdplace
+++ b/pretrained/holdplace
--- a/test.ipynb
+++ b/test.ipynb