Hand over face (High Resolution Masks) about hand-segmentation-in-the-wild HOT 11 CLOSED

`import xmltodict
import glob
import os
from PIL import Image
from PIL import Image, ImageDraw
import argparse
from tqdm import tqdm


parser = argparse.ArgumentParser(description='simple parser')

parser.add_argument('--input_folder', type=str, default="./xml/",
                    help='path to only xml files', required=False)

parser.add_argument('--output_folder', type=str, default="output/", help='path to store the images ', required=False)
args = parser.parse_args()


def draw(image, pt_dict):
  pt_arr = []
  for p in pt_dict:
    pt_arr.append((int(p["x"]),int(p["y"])))
  image_draw = ImageDraw.Draw(image)
  image_draw.polygon(pt_arr, fill="#FFFFFF", outline ="white")
  del image_draw
  return image


def run_prog(in_folder, out_folder):
  #print(in_folder, out_folder)
  my_files = sorted(glob.glob(os.path.join(in_folder, "*.xml")))
  #print(my_files)
  for f in tqdm(my_files):
      try:
        if f.split(".")[-1].lower() not in "xml":
          print(f.split(".")[-1].lower())
          continue
        with open(f, 'r') as file:
          data = file.read().replace('\n', '')
          my_dict = xmltodict.parse(data)
          #print(data)
        #print(data)

        #print(my_dict)
        fname=my_dict['annotation']["filename"]
        fname_no = int(fname.replace(".jpg", ""))
        img_size_w =  int(my_dict['annotation']["imagesize"]["nrows"])
        img_size_h =  int(my_dict['annotation']["imagesize"]["ncols"])
        my_object = my_dict['annotation']['object']

        image = Image.new('RGB', (img_size_h, img_size_w))

        if  isinstance(my_object, dict):
          poly_pt_dict=my_object['polygon']
          pt_dict = poly_pt_dict['pt']
          image = draw(image, pt_dict)
        elif isinstance(my_object, list):
          #print(type(my_object))
          for el in my_object:
            #print(el)
            poly_pt_dict = el['polygon']
            pt_dict = poly_pt_dict['pt']
            image = draw(image, pt_dict)
            # pass
          #poly_pt_dict = my_object[-1]
          #continue
        #print(my_object)
        #poly_pt_dict=my_object['polygon']
        #print(poly_pt_dict)

        #print(pt_dict)


        #display(image)
        xx = (os.path.join(out_folder, f"{str(fname_no)}.jpg"))

        xx = xx[23:]
        #print(xx)

        test = './output2/' + xx
        xx = './images_original_size/' + xx
        print('xx',xx)
        #print(test)
        #break
        #print(xx,test)
        #print('xx',xx)
        #img = Image.open(xx)
        #img.save(test)
        os.rename(xx,test)
        image.save(os.path.join(out_folder, f"{str(fname_no)}.png"))
        #del image_draw
        del image
        del my_dict

      except:
          continue

if __name__ == "__main__":
  run_prog(args.input_folder, args.output_folder)