Comments by Fatema Shawki (Top 3 by date)

The dataset consists of 4 classes, each class consists of .jpg files and .txt files (contains the locations and class of the object in yolo format)
I'm sorry but I'm currently unclear on the implications of what you are suggesting. Could you please provide more information or clarification so that I can better understand?

I used the export function to convert from yolo to tensor flow and then converted the tensor flow saved model to tensor flow lite using tf.convertor
1)
!python /content/yolov5/export.py --weights /content/yolov5/runs/train/exp/weights/best.pt --include tflite --img 640 --data data.yaml
2)
converter = tf.lite.TFLiteConverter.from_saved_model('/content/yolov5/runs/train/exp/weights/best_saved_model')
tflite_model = converter.convert()
with open('model.tflite', 'wb') as f:
f.write(tflite_model)