def setUp(self):
super(CLEVRDataClosestTest, self).setUp(
data_wrapper=clevr.CLEVRData(task="closest_object_distance"),
num_classes=6,
expected_num_samples=dict(
train=63000,
val=7000,
trainval=70000,
test=15000,
),
required_tensors_shapes={
"image": (None, None, 3),
"label": (),
},
tfds_label_key_map={})
def setUp(self):
super(CLEVRDataCountCylindersTest,
self).setUp(data_wrapper=clevr.CLEVRData(task="count_cylinders"),
num_classes=11,
expected_num_samples=dict(
train=63000,
val=7000,
trainval=70000,
test=15000,
),
required_tensors_shapes={
"image": (None, None, 3),
"label": (),
},
tfds_label_key_map={})