def setUp(self):
super(KittiDataCountNearTest, self).setUp(
data_wrapper=kitti.KittiData(task="count_near"),
num_classes=16,
expected_num_samples=dict(
train=5077,
val=634,
trainval=5711,
test=634,
),
required_tensors_shapes={
"image": (None, None, 3),
"label": (),
},
tfds_label_key_map={})
def setUp(self):
super(KittiDataClosestVehicleTest, self).setUp(
data_wrapper=kitti.KittiData(task="closest_vehicle_distance"),
num_classes=4,
expected_num_samples=dict(
train=5077,
val=634,
trainval=5711,
test=634,
),
required_tensors_shapes={
"image": (None, None, 3),
"label": (),
},
tfds_label_key_map={})
def setUp(self):
super(KittiDataCountFarTest, self).setUp(
data_wrapper=kitti.KittiData(task="count_far"),
num_classes=16,
expected_num_samples=dict(
train=6347,
val=423,
trainval=6770,
test=711,
train800val200=1000,
train800=800,
val200=200,
),
required_tensors_shapes={
"image": (None, None, 3),
"label": (),
},
tfds_label_key_map={})
