# cat check_no_label_jpg.py
import os
import shutil
tf_dir = './JPEG_half'
label_dir = './label'
r = []
no_label_jpg = "./no_label_jpg/"
#tf_list = [os.path.join(tf_dir,f) for f in os.listdir(tf_dir) ]
#print("jpg file is {}".format(tf_list))
tf_name = [ f for f in os.listdir(tf_dir) ]
#print("jpg file name is {}".format(tf_name))
print("jpg files are {}".format(len(tf_name)))
label_name = [ f.replace('txt','jpg') for f in os.listdir(label_dir) ]
#print("label file name is {}".format(label_name))
print("label files are {}".format(len(label_name)))
set1 = set(tf_name)
set2 = set(label_name)
z = set(set1^set2)
#print("below has no label: {}".format(z))
print("{} have no labels".format(len(z)))
for i in z:
    r = os.path.join(tf_dir,i)
   # print(r)
    shutil.move(r, no_label_jpg)