dima806/headgear_image_detection

🤗 Hugging Face 来源image-classificationapache-2.086M 参数343 MBsafetensors✓ 13 个校验和今天更新

一条命令提交

在你的模型文件夹旁边运行它。它会制作种子、将文件与 Hugging Face 比对，然后提交。你只需开始做种，并粘贴你账户中的密钥。它只读取你的文件，绝不修改。如果愿意，可以先阅读脚本。

curl -fsSL https://pirateface.co/package.sh | bash -s -- --repo dima806/headgear_image_detection ./model-folder

Returns headgear type given an image.

See https://www.kaggle.com/code/dima806/headgear-image-detection-vit for more details.

Classification report:

                 precision    recall  f1-score   support

          BERET     1.0000    0.9565    0.9778       115
         FEDORA     0.9913    1.0000    0.9956       114
        SOMBERO     1.0000    1.0000    1.0000       115
       HARD HAT     1.0000    1.0000    1.0000       115
            FEZ     1.0000    0.9912    0.9956       114
      ZUCCHETTO     1.0000    0.9912    0.9956       114
        TOP HAT     1.0000    1.0000    1.0000       115
    DEERSTALKER     0.9913    1.0000    0.9956       114
      ASCOT CAP     0.9500    1.0000    0.9744       114
       PORK PIE     0.9739    0.9825    0.9782       114
MILITARY HELMET     1.0000    1.0000    1.0000       115
        BICORNE     1.0000    0.9912    0.9956       114
FOOTBALL HELMET     1.0000    1.0000    1.0000       115
     MOTARBOARD     0.9913    1.0000    0.9956       114
         BOATER     1.0000    1.0000    1.0000       115
    PITH HELMET     0.9913    1.0000    0.9956       114
    SOUTHWESTER     1.0000    0.9912    0.9956       114
         BOWLER     0.9912    0.9825    0.9868       114
   GARRISON CAP     1.0000    0.9912    0.9956       114
   BASEBALL CAP     1.0000    1.0000    1.0000       115

       accuracy                         0.9939      2288
      macro avg     0.9940    0.9939    0.9939      2288
   weighted avg     0.9940    0.9939    0.9939      2288