ACGPN(Adaptive Content Generating and Preserving Network )

import gdown

import numpy as np

from PIL import Image

import IPython

import gdown

import os

import sys

from predict_pose import generate_pose_keypoints

sys.path.append('/users/cbf107/cbf107020/ACGPN-master/U_2_Net')

import u2net_run

import u2net_load

u2net = u2net_load.model(model_name='u2netp')

f = open('return.txt','r')

cloth_number = int(f.readlines()[0])

print(cloth_number)

cloth_name = '000001_1.png'

cloth_path = os.path.join('inputs/cloth', sorted(os.listdir('inputs/cloth'))[cloth_number])

cloth = Image.open(cloth_path)

cloth = cloth.resize((192, 256), Image.BICUBIC).convert('RGB')

cloth.save(os.path.join('Data_preprocessing/test_color', cloth_name))

u2net_run.infer(u2net, 'Data_preprocessing/test_color' , 'Data_preprocessing/test_edge')

import time

start_time = time.time()

img_name = '000001_0.png'

img_path = os.path.join('inputs/img', sorted(os.listdir('inputs/img'))[0])

img = Image.open(img_path)

img = img.resize((192,256), Image.BICUBIC)

img_path = os.path.join('Data_preprocessing/test_img', img_name)

img.save(img_path)

resize_time = time.time()

print('Resized image in {}s'.format(resize_time-start_time))

os.system("python3 Self-Correction-Human-Parsing-for-ACGPN/simple_extractor.py --dataset 'lip' --model-restore 'lip_final.pth' --input-dir 'Data_preprocessing/test_img' --output-dir 'Data_preprocessing/test_label'")

parse_time = time.time()

print('Parsing generated in {}s'.format(parse_time-resize_time))

pose_path = os.path.join('Data_preprocessing/test_pose', img_name.replace('.png', '_keypoints.json'))

generate_pose_keypoints(img_path, pose_path)

pose_time = time.time()

print('Pose map generated in {}s'.format(pose_time-parse_time))

os.system("rm -rf Data_preprocessing/test_pairs.txt")

with open('Data_preprocessing/test_pairs.txt','w') as f:

    f.write('000001_0.png 000001_1.png')

os.system("python test.py")

os.system("rm -rf Data_preprocessing/test_color/*")

os.system("rm -rf Data_preprocessing/test_colormask/*")

os.system("rm -rf Data_preprocessing/test_edge/*")

os.system("rm -rf Data_preprocessing/test_img/*")

os.system("rm -rf Data_preprocessing/test_label/*")

os.system("rm -rf Data_preprocessing/test_mask/*")

os.system("rm -rf Data_preprocessing/test_pose/*")

#os.system("rm -rf inputs/cloth/*")

os.system("rm -rf inputs/img/*")