mazx4960 / weapon-detection

weapon detection

Results

Resnet50

10 epochs

Accuracy of the model on the test images: 66.92%
F1 score: 0.57
Confusion matrix: 
[[933 359]
 [326 453]]

Resnet101, 10 epochs

Accuracy of the model on the test images: 68.66%
F1 score: 0.57
Confusion matrix:
tensor([[999, 293],
        [356, 423]])