archit11
/

videomae-base-finetuned-ucfcrime-full

Video Classification

Generated from Trainer

vandalism-dectection

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

archit11 commited on Mar 18

Commit

d2a2733

•

1 Parent(s): 35a8c91

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -20,7 +20,8 @@ should probably proofread and complete it, then remove this comment. -->
 # videomae-base-finetuned-ucfcrime-full2
-This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
 - Loss: 2.5014
 - Accuracy: 0.225
@@ -101,7 +102,7 @@ with torch.no_grad():
     outputs = model(**inputs)
     logits = outputs.logits
-# model predicts one of the 400 Kinetics-400 classes
 predicted_label = logits.argmax(-1).item()
 print(model.config.id2label[predicted_label])
 ```

 # videomae-base-finetuned-ucfcrime-full2
+This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on the [UCF-CRIME](https://paperswithcode.com/dataset/ucf-crime)
+dataset.
 It achieves the following results on the evaluation set:
 - Loss: 2.5014
 - Accuracy: 0.225
     outputs = model(**inputs)
     logits = outputs.logits
+# model predicts one of the 13 ucf-crime classes
 predicted_label = logits.argmax(-1).item()
 print(model.config.id2label[predicted_label])
 ```