ChatSam / Weapons_detection_immich

Weapons detection feature based on the immich self-hosted photo and video hosting platform.

Home Page:https://immich.app

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Improve Accuracy of Initial Weapon Detection Model on Images

RuchiGupta20 opened this issue · comments

Currently, our initial weapon detection model on images requires enhancements to achieve higher accuracies. Although the model is functional, it lacks precision and sometimes misclassifies objects, leading to potential safety concerns. To address this issue, we need to undertake improvements and optimizations.

Worked with a more robust dataset of knives and guns combined with the original training, testing, and validation dataset that resulted in better model performance.