""" Author: Soubhik Sanyal Copyright (c) 2019, Soubhik Sanyal All rights reserved. Max-Planck-Gesellschaft zur Foerderung der Wissenschaften e.V. (MPG) is holder of all proprietary rights on this computer program. You can only use this computer program if you have closed a license agreement with MPG or you get the right to use the computer program from someone who is authorized to grant you that right. Any use of the computer program without a valid license is prohibited and liable to prosecution. Copyright 2019 Max-Planck-Gesellschaft zur Foerderung der Wissenschaften e.V. (MPG). acting on behalf of its Max Planck Institute for Intelligent Systems and the Max Planck Institute for Biological Cybernetics. All rights reserved. More information about RingNet is available at https://ringnet.is.tue.mpg.de. based on github.com/akanazawa/hmr """ # Preprocessing. import numpy as np import cv2 def resize_img(img, scale_factor): new_size = (np.floor(np.array(img.shape[0:2]) * scale_factor)).astype(int) new_img = cv2.resize(img, (new_size[1], new_size[0])) # This is scale factor of [height, width] i.e. [y, x] actual_factor = [ new_size[0] / float(img.shape[0]), new_size[1] / float(img.shape[1]) ] return new_img, actual_factor def scale_and_crop(image, scale, center, img_size): image_scaled, scale_factors = resize_img(image, scale) # Swap so it's [x, y] scale_factors = [scale_factors[1], scale_factors[0]] center_scaled = np.round(center * scale_factors).astype(np.int) margin = int(img_size / 2) image_pad = np.pad( image_scaled, ((margin, ), (margin, ), (0, )), mode='edge') center_pad = center_scaled + margin # figure out starting point start_pt = center_pad - margin end_pt = center_pad + margin # crop: crop = image_pad[start_pt[1]:end_pt[1], start_pt[0]:end_pt[0], :] proc_param = { 'scale': scale, 'start_pt': start_pt, 'end_pt': end_pt, 'img_size': img_size } return crop, proc_param