new_contents: import os
  
  import numpy as np
  
  data_dir = "data/dataset/"
  jpg_filenames = list(filter(lambda x: x[-3:] == "jpg", os.listdir(data_dir)))
  
  # Randomly select the test dataset
  test_percentage = 0.1
  n_test = round(len(jpg_filenames) * test_percentage)
  if n_test == 0: n_test = 1
  
  # Randomly select the images for testing
  test_indexes = np.random.choice(len(jpg_filenames), n_test, replace=False)
  
  jpg_filenames_copy = jpg_filenames[:]
  
  with open("test.txt", "w") as f:
      for index in test_indexes:
          # Write filename
          f.write(data_dir + jpg_filenames[index] + "\n")
          # Remove from copy list
          jpg_filenames_copy.pop(index)
  
  # Write from the copy list
  with open("train.txt", "w") as f:
      for filename in jpg_filenames_copy:
          f.write(data_dir + filename + "\n")
  
subject: Add script to split the dataset