import os
from shutil import copyfile
import numpy as np

dir_name_first = './imagens/Img_Inteira/mao_aberta'
dir_name_secon = './imagens/f1'
ims = os.listdir(dir_name_first)

total_array = np.zeros(1000)#np.arange(len(ims))
np.random.shuffle(total_array)

seq_train = total_array[0: int(len(total_array) * (9/10))]
seq_test = total_array[len(total_array) - int(len(total_array) * (1/10)): len(total_array)]

print(len(seq_train))
print(len(seq_test))

for key, val in enumerate(seq_train):
	file_name = sorted(os.listdir(dir_name_first), key=len)[val]
	copyfile(os.path.join(dir_name_first, file_name), os.path.join(dir_name_secon ,'train', 'm1', file_name))

for key, val in enumerate(seq_test):
	file_name = sorted(os.listdir(dir_name_first), key=len)[val]
	copyfile(os.path.join(dir_name_first, file_name), os.path.join(dir_name_secon ,'test', 'm1', file_name))