if __name__ == "__main__":
import sys
sys.path.append("../")
# import models
from PIL import Image, ImageOps
import imageio
import torchvision.transforms
import os
decomposer = Decomposer()
decomposer.load_state_dict(torch.load("saved/decomposer/state.t7"))
shader = Shader()
shader.load_state_dict(torch.load("saved/shader/state.pth"))
composer_path = "/home/ab2383/intrinsics-network/saved/composer/state.t7"
composer = Composer(decomposer, shader)
composer = composer.cuda()
composer.load_state_dict(torch.load(composer_path))
print("Composer Built")
# folders = ["00277", "00339", "00267"]
# base_path = "/phoenix/S3/ab2383/data/TikTok_dataset/"
save_path = "/home/ab2383/intrinsics-network/"
# for folder in folders:
# imgs_path = base_path + folder + "/images/"
# masks_path = base_path + folder + "/masks/"
# for img in os.listdir(imgs_path):
base_path = "/home/ab2383/intrinsics-network/dataset/output/motorbike_train/"
