This is my code for loading data from csv
 from tflite_model_maker.object_detector import DataLoader
  train_dir = '/content/drive/MyDrive/Training_demo/images/my_dataset/train'
train_data, validation_data, test_data = DataLoader.from_csv('/content/drive/MyDrive/Training_demo/annotations/train_dataset.csv', images_dir = train_dir)