许可证：apache-2.0
标签：

• 图像分类
• 视觉
• 训练生成
  指标：
• 准确率

模型索引：

• 名称：outputs
  结果：
  • 任务：
    名称：图像分类
    类型：image-classification
    指标：
    • 名称：准确率
      类型：accuracy
      值：0.9107332624867163

outputs

该模型是基于microsoft/beit-base-patch16-224-pt22k-ft22k在PETA数据集上微调的版本。
在评估集上取得了以下结果：

• 损失：0.2170
• 准确率：0.9107

模型描述

需补充更多信息

使用方法

可通过Transformers的pipeline调用该模型：

from transformers import pipeline  
gender_classifier = pipeline(model="NTQAI/pedestrian_gender_recognition")  
image_path = "abc.jpg"  

results = gender_classifier(image_path)  
print(results)  

预期用途与限制

需补充更多信息

训练与评估数据

需补充更多信息

训练流程

训练超参数

训练过程中使用的超参数如下：

• 学习率：2e-05
• 训练批次大小：8
• 评估批次大小：8
• 随机种子：1337
• 优化器：Adam（beta1=0.9，beta2=0.999，epsilon=1e-08）
• 学习率调度器类型：linear
• 训练轮次：5.0

训练结果

框架版本

• Transformers 4.24.0.dev0
• Pytorch 1.12.1+cu113
• Datasets 2.6.1
• Tokenizers 0.13.1

联系方式

关于本项目的个人沟通，请联系Nha Nguyen Van（nha282@gmail.com）。