yixiong/FoodClassifier
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

不要类别加权

Browse Source
This commit is contained in:
zhangpu
2025-11-26 14:50:00 +08:00
parent 99d9eea757
commit 0e8a1df590
5 changed files with 28 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train/grid_search_cosface.py
+9 -3
View File
@@ -41,10 +41,16 @@ logging.basicConfig(
logger = logging.getLogger(__name__)
# ==================== 网格搜索空间定义 ====================
"""
'dish': {
's': [56.0, 60.0, 64.0, 68.0], # scale参数
'm': [0.32, 0.35, 0.38, 0.40], # margin参数
},
"""
GRID_PARAMS = {
'dish': {
's': [56.0, 60.0, 64.0, 68.0], # scale参数
'm': [0.32, 0.35, 0.38, 0.40], # margin参数
'm': [0.32, 0.35, 0.38, 0.40,0.45,0.50], # margin参数
},
'whole_ingredient': {
's': [56.0, 60.0, 64.0, 68.0],
@@ -354,8 +360,8 @@ def grid_search_main(
if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser(description='CosFace超参数网格搜索')
parser.add_argument('--task', choices=list(TASKS.keys()), default='dish', help='任务名称')
# parser.add_argument('--task', choices=list(TASKS.keys()), default='whole_ingredient', help='任务名称')
# parser.add_argument('--task', choices=list(TASKS.keys()), default='dish', help='任务名称')
parser.add_argument('--task', choices=list(TASKS.keys()), default='whole_ingredient', help='任务名称')
parser.add_argument('--max_configs', type=int, default=None, help='最大配置数（用于测试）')
parser.add_argument('--epochs', type=int, default=100, help='每个配置的最大训练轮数')
parser.add_argument('--patience', type=int, default=10, help='早停容忍轮数')