8 PM – Advanced AI/ML – Joel

Today we did:

1. We continued with the Glasses Detection project.

Homework:

1. If you haven’t already put the best.pt into the Pycharm folder, make sure that you finish training first, and once training finishes, open the runs folder -> detect -> train -> weights -> and find best.pt. Then, right-click on that best.pt file, and click download. It might take some time. Once you’ve downloaded it, then move it into the Pycharm project and rename it to glasses.pt.
2. After that, make sure to change the title and everything to make it for the glasses project, and also remove these lines of code:
   cls = int(box.cls[0])
   label = coco_classes[cls]
3. Then, make sure that you check for the confidence value. If it is greater than 0.8, then and only then draw the bounding box and confidence/label text.
4. When you test this code by running streamlit run yolo_glasses_detection.py, and you see that it doesn’t detect glasses, try decreasing the confidence threshold to something like 0.4. Play around with different confidence levels to see how it affects the model prediction.