-
The .ipynb file contains the full code for training the model and testing it, and also exporting it as well. You do not need to run it anymore since we have already trained the model.
-
The yolov5 folder contains the full yolo model dependencies. Inside it, there is a 'runs' folder which contains all training and evaluation metrics for the model.
-
Inside the 'yolov5/runs/train/weights, we have best-fp16.tflite model. It is the yolo model weights exported in tflite format which you can use in your Android studio.
-
For retraining, download the dataset from this link "https://drive.google.com/drive/folders/18U0gQetGk9y2s-UF7RaHBKSA-EdTk3Zr?usp=drive_link"
-
The report.doc file contains the report for the model training and evaluation.