Add files via upload

Author: Camebush

object.data

@@ -0,0 +1,5 @@
+classes = 2
+train = data/train.txt
+valid = data/test.txt
+names = data/object.names
+backup = /content/drive/MyDrive/Yolov4/training

object.names

@@ -0,0 +1,2 @@
+Samuel Eto'o
+Francis Ngannou

process.py

@@ -0,0 +1,26 @@
+import os
+import numpy as np
+# Current directory
+current_dir = 'data/object'
+
+# Percentage of images to be used for the test set
+test_ratio = 0.15
+
+# Valide image files we will use
+valid_extensions = [".jpg",".gif",".png",".jpeg"]
+
+allfilenames = [filename for filename in os.listdir(current_dir) if os.path.splitext(filename)[1].lower() in valid_extensions]
+np.random.shuffle(allfilenames)
+
+train_FileNames , test_FileNames = np.split(np.array(allfilenames), [int(len(allfilenames)*(1-test_ratio))])
+
+train_FileNames = [current_dir + '/' + name for name in train_FileNames.tolist()]
+test_FileNames = [current_dir + '/' + name for name in test_FileNames.tolist()]
+
+with open('data/train.txt', 'w') as train_f:
+    for name in train_FileNames:
+        train_f.write(name+'\n')
+
+with open('data/test.txt', 'w') as test_f:
+    for name in test_FileNames:
+        test_f.write(name+'\n')

rename.py

@@ -0,0 +1,16 @@
+import os
+from PIL import Image
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument('--dir', help='Location to run script',required=True)
+
+args = parser.parse_args()
+
+count = 0
+
+for i in os.listdir(args.dir):
+	i_path = os.path.join(args.dir, i)
+	new_i_path = os.path.join(args.dir,'img-'+str(count)+'.'+i.split('.')[-1])
+	os.rename(i_path,new_i_path)
+	count+=1