Import csv
import csv
Import os
import os
import shutil
Main Function
def main():
# Open dataset file
dataset = open('dataset_10000.csv', newline='')
# Initialize csvreader for dataset
reader = csv.reader(dataset)
# Read data from reader
data = list(reader)
# Variables for progress counter
lines = len(data)
i = 0
# Analyze data in dataset
for row in data:
# Assign image name and state to variables
image = row[0] + '.jpeg'
state = row[1]
# Print image information
print('({}/{}) Processing image ({}): {}'.format(i + 1, lines, state, image))
# Increment i
i += 1
# Determine action to perform
if state is '0':
# Attempt to move the file
try:
# Move the file to folderone/
#os.rename(image, 'no/' + image)
shutil.copyfile(image, 'no/' + image)
# Inform the user of action being taken
print(' -> Moved to folderone/')
except FileNotFoundError:
# Inform the user of the failure
print(' -> Failed to find file')
elif state in ['1', '2', '3', '4']:
# Attempt to move the file
try:
# Move the file to foldertwo/
#os.rename(image, 'yes/' + image)
#shutil.copyfile(image,'yes/' + image)
shutil.copyfile(image, 'yes/' + image)
# Inform the user of action being taken
print(' -> Moved to foldertwo/')
except FileNotFoundError:
# Inform the user of the failure
print(' -> Failed to find file')
Execute main function if name is equal to main
if name == 'main':
main()
