dima806's picture
Update README.md
c3c8d0a verified
metadata
license: apache-2.0
metrics:
  - accuracy
  - f1
base_model:
  - google/vit-base-patch16-224-in21k
pipeline_tag: image-classification

Returns with about 83% accuracy whether is a crime in a CCTV camera image.

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

image/png

Classification report:

              precision    recall  f1-score   support

       Crime     0.7971    0.7823    0.7896     46356
      Normal     0.8467    0.8579    0.8522     64952

    accuracy                         0.8264    111308
   macro avg     0.8219    0.8201    0.8209    111308
weighted avg     0.8260    0.8264    0.8262    111308