中文Sign up

dima806/headgear_image_detection

🤗 Hugging Face sourceimage-classificationapache-2.086M params343 MBsafetensors✓ 13 checksumsupdated today
Submit in one command

Run it next to your model folder. It makes the torrent, checks your files against Hugging Face, and submits it. You just start seeding and paste your key from your account. It only reads your files and never changes them. Read the script first if you like.

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

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