files for creating train and testdataset

Hi All I am not the authors of of these two files authors can reachout to me

Author: jakkcoder

creating-files-data-and-name.py

@@ -0,0 +1,104 @@
+
+
+"""
+Course:  Training YOLO v3 for Objects Detection with Custom Data
+
+Section-3
+Labelling new Dataset in YOLO format
+File: creating-files-data-and-name.py
+"""
+
+
+# Creating files labelled_data.data and classes.names
+# for training in Darknet framework
+#
+# Algorithm:
+# Setting up full paths --> Reading file classes.txt -->
+# --> Creating file classes.names -->
+# --> Creating file labelled_data.data
+#
+# Result:
+# Files classes.names and labelled_data.data needed to train
+# in Darknet framework
+
+
+"""
+Start of:
+Setting up full path to directory with labelled images
+"""
+
+# Full or absolute path to the folder with images
+# Find it with Py file getting-full-path.py
+# Pay attention! If you're using Windows, yours path might looks like:
+# r'C:\Users\my_name\Downloads\video-to-annotate'
+# or:
+# 'C:\\Users\\my_name\\Downloads\\video-to-annotate'
+full_path_to_images = '/home/my_name/Downloads/video-to-annotate'
+
+"""
+End of:
+Setting up full path to directory with labelled images
+"""
+
+
+"""
+Start of:
+Creating file classes.names
+"""
+
+# Defining counter for classes
+c = 0
+
+# Creating file classes.names from existing one classes.txt
+# Pay attention! If you're using Windows, it might need to change
+# this: + '/' +
+# to this: + '\' +
+# or to this: + '\\' +
+with open(full_path_to_images + '/' + 'classes.names', 'w') as names, \
+     open(full_path_to_images + '/' + 'classes.txt', 'r') as txt:
+
+    # Going through all lines in txt file and writing them into names file
+    for line in txt:
+        names.write(line)  # Copying all info from file txt to names
+
+        # Increasing counter
+        c += 1
+
+"""
+End of:
+Creating file classes.names
+"""
+
+
+"""
+Start of:
+Creating file labelled_data.data
+"""
+
+# Creating file labelled_data.data
+# Pay attention! If you're using Windows, it might need to change
+# this: + '/' +
+# to this: + '\' +
+# or to this: + '\\' +
+with open(full_path_to_images + '/' + 'labelled_data.data', 'w') as data:
+    # Writing needed 5 lines
+    # Number of classes
+    # By using '\n' we move to the next line
+    data.write('classes = ' + str(c) + '\n')
+
+    # Location of the train.txt file
+    data.write('train = ' + full_path_to_images + '/' + 'train.txt' + '\n')
+
+    # Location of the test.txt file
+    data.write('valid = ' + full_path_to_images + '/' + 'test.txt' + '\n')
+
+    # Location of the classes.names file
+    data.write('names = ' + full_path_to_images + '/' + 'classes.names' + '\n')
+
+    # Location where to save weights
+    data.write('backup = backup')
+
+"""
+End of:
+Creating file labelled_data.data
+"""

creating-train-and-test-txt-files.py

@@ -0,0 +1,124 @@
+
+
+"""
+Course:  Training YOLO v3 for Objects Detection with Custom Data
+
+Section-3
+Labelling new Dataset in YOLO format
+File: creating-train-and-test-txt-files.py
+"""
+
+
+# Creating files train.txt and test.txt
+# for training in Darknet framework
+#
+# Algorithm:
+# Setting up full paths --> List of paths -->
+# --> Extracting 15% of paths to save into test.txt file -->
+# --> Writing paths into train and test txt files
+#
+# Result:
+# Files train.txt and test.txt with full paths to images
+
+
+# Importing needed library
+import os
+
+
+"""
+Start of:
+Setting up full path to directory with labelled images
+"""
+
+# Full or absolute path to the folder with images
+# Find it with Py file getting-full-path.py
+# Pay attention! If you're using Windows, yours path might looks like:
+# r'C:\Users\my_name\Downloads\video-to-annotate'
+# or:
+# 'C:\\Users\\my_name\\Downloads\\video-to-annotate'
+full_path_to_images = '/home/my_name/Downloads/video-to-annotate'
+
+"""
+End of:
+Setting up full path to directory with labelled images
+"""
+
+
+"""
+Start of:
+Getting list of full paths to labelled images
+"""
+
+# Check point
+# Getting the current directory
+# print(os.getcwd())
+
+# Changing the current directory
+# to one with images
+os.chdir(full_path_to_images)
+
+# Check point
+# Getting the current directory
+# print(os.getcwd())
+
+# Defining list to write paths in
+p = []
+
+# Using os.walk for going through all directories
+# and files in them from the current directory
+# Fullstop in os.walk('.') means the current directory
+for current_dir, dirs, files in os.walk('.'):
+    # Going through all files
+    for f in files:
+        # Checking if filename ends with '.jpeg'
+        if f.endswith('.jpeg'):
+            # Preparing path to save into train.txt file
+            # Pay attention!
+            # If you're using Windows, it might need to change
+            # this: + '/' +
+            # to this: + '\' +
+            # or to this: + '\\' +
+            path_to_save_into_txt_files = full_path_to_images + '/' + f
+
+            # Appending the line into the list
+            # We use here '\n' to move to the next line
+            # when writing lines into txt files
+            p.append(path_to_save_into_txt_files + '\n')
+
+
+# Slicing first 15% of elements from the list
+# to write into the test.txt file
+p_test = p[:int(len(p) * 0.15)]
+
+# Deleting from initial list first 15% of elements
+p = p[int(len(p) * 0.15):]
+
+"""
+End of:
+Getting list of full paths to labelled images
+"""
+
+
+"""
+Start of:
+Creating train.txt and test.txt files
+"""
+
+# Creating file train.txt and writing 85% of lines in it
+with open('train.txt', 'w') as train_txt:
+    # Going through all elements of the list
+    for e in p:
+        # Writing current path at the end of the file
+        train_txt.write(e)
+
+# Creating file test.txt and writing 15% of lines in it
+with open('test.txt', 'w') as test_txt:
+    # Going through all elements of the list
+    for e in p_test:
+        # Writing current path at the end of the file
+        test_txt.write(e)
+
+"""
+End of:
+Creating train.txt and test.txt files
+"""