Click here to Skip to main content
15,844,754 members

Comments by Fatema Shawki (Top 3 by date)

Fatema Shawki 14-Apr-23 10:11am View    
Is there anything else you require?
Fatema Shawki 23-Mar-23 8:39am View    
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?
Fatema Shawki 20-Mar-23 19:43pm View    
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)