Quantcast
Channel: Raspberry Pi Forums
Viewing all articles
Browse latest Browse all 8023

General programming discussion • Re: Retraining AI model for image detection

$
0
0
Hi! MobileNet-SSD is fast but limited in accuracy for new objects. To improve it:

Collect & label images – use LabelImg
in VOC format.
Set up training – better on a PC with GPU, install Caffe.
Modify model – change num_classes in .prototxt to match your objects.
Fine-tune – start from pre-trained weights (MobileNetSSD_deploy.caffemodel) and train with your data.
Test – load new model in OpenCV using cv2.dnn.readNetFromCaffe().
Tip: start small (2–3 classes) before scaling up.

Statistics: Posted by Akash Jana — Sat Nov 15, 2025 5:03 am



Viewing all articles
Browse latest Browse all 8023

Trending Articles