File: /home/ekspardev/onarbeni-master/tools/arrange_database.py
import os
import glob
import shutil
pwd=os.getcwd()
pwd
possitive = glob.glob('data/cropped/possitive/*.*')
negative = glob.glob('data/cropped/negative/*.*')
len(possitive), len(negative)
possitive_train = possitive[:int(len(possitive)*0.8)]
negative_train = negative[:int(len(negative)*0.8)]
possitive_test = possitive[int(len(possitive)*0.8):]
negative_test = negative[int(len(negative)*0.8):]
for i, image in enumerate(possitive_train):
shutil.copy(image, image.replace('cropped', 'cropped/train'))
for i, image in enumerate(negative_train):
shutil.copy(image, image.replace('cropped', 'cropped/train'))
for i, image in enumerate(possitive_test):
shutil.copy(image, image.replace('cropped', 'cropped/test'))
for i, image in enumerate(negative_test):
shutil.copy(image, image.replace('cropped', 'cropped/test'))