def __init__(self, cfg):
self.cfg = cfg
self.data = self.load_dataset()
self.num_images = len(self.data)
self.max_input_sizesquare = cfg.get("max_input_size", 1500) ** 2
self.min_input_sizesquare = cfg.get("min_input_size", 64) ** 2
self.locref_scale = 1.0 / cfg['locref_stdev']
self.stride = cfg['stride']
self.half_stride = self.stride / 2
self.scale = cfg['global_scale']
self.scale_jitter_lo = cfg.get("scale_jitter_lo", 0.75)
self.scale_jitter_up = cfg.get("scale_jitter_up", 1.25)
self.cfg['crop'] = cfg.get("crop", True)
self.cfg['cropratio'] = cfg.get("cropratio", 0.4)
# what is the minimal frames size for cropping plus/minus ie.. [-100,100]^2 for an arb. joint
self.cfg['minsize'] = cfg.get("minsize", 100)
self.cfg['leftwidth'] = cfg.get("leftwidth", 400)
self.cfg['rightwidth'] = cfg.get("rightwidth", 400)
self.cfg['topheight'] = cfg.get("topheight", 400)
self.cfg['bottomheight'] = cfg.get("bottomheight", 400)
if self.cfg['mirror']:
self.symmetric_joints = mirror_joints_map(cfg['all_joints'], cfg['num_joints'])
self.curr_img = 0
self.set_shuffle(cfg['shuffle'])
def __init__(self, cfg):
self.cfg = cfg
self.data = self.load_dataset()
self.num_images = len(self.data)
if self.cfg.mirror:
self.symmetric_joints = mirror_joints_map(cfg.all_joints, cfg.num_joints)
self.curr_img = 0
self.set_shuffle(cfg.shuffle)
def __init__(self, cfg):
self.cfg = cfg
self.data = self.load_dataset()
self.num_images = len(self.data)
if self.cfg["mirror"]:
self.symmetric_joints = mirror_joints_map(cfg["all_joints"],
cfg["num_joints"])
self.curr_img = 0
self.set_shuffle(cfg["shuffle"])
def __init__(self, cfg):
self.cfg = cfg
self.data = self.load_dataset()
self.num_images = len(self.data)
if self.cfg['mirror']:
self.symmetric_joints = mirror_joints_map(cfg['all_joints'],
cfg['num_joints'])
self.curr_img = 0
self.set_shuffle(cfg['shuffle'])
def __init__(self, cfg):
self.cfg = cfg
self.data = self.load_dataset()
self.num_images = len(self.data)
self.max_input_sizesquare = cfg.get("max_input_size", 1500) ** 2
self.min_input_sizesquare = cfg.get("min_input_size", 64) ** 2
self.locref_scale = 1.0 / cfg.locref_stdev
self.stride = cfg.stride
self.half_stride = cfg.stride / 2
self.scale = cfg.global_scale
self.scale_jitter_lo = cfg.get("scale_jitter_lo", 0.75)
self.scale_jitter_up = cfg.get("scale_jitter_up", 1.25)
if self.cfg.mirror:
self.symmetric_joints = mirror_joints_map(cfg.all_joints, cfg.num_joints)
self.curr_img = 0
self.set_shuffle(cfg.shuffle)
