hom_traj_gen/ml_lib/utils/transforms.py
2020-04-08 14:50:16 +02:00

12 lines
240 B
Python

import numpy as np
class AsArray(object):
def __init__(self, width, height):
self.width = width
self.height = height
def __call__(self, x):
array = np.zeros((self.width, self.height))
return array