Making a robust and secure IoT edge device framework for my senior project class.
-
Download the dataset from Kaggle.
-
Place the dataset into the repo and run the convert_to_yolo.py script. There may be file pathing errors so ensure the dataset paths are as described in the python script editing either the path used in the python file or of the dataset. From here you can download your yolo versions weights from ultralytics and run the following:
yolo train model=\<your model weights\> data=face.yaml imgsz=640 epochs=100 batch=16

