def get_test_image_dataset(self, sample_type):
return RandomImageBinaryClassDataset(
crop_size=224,
class_ratio=0.5,
num_samples=100,
seed=0,
sample_type=sample_type,
)
def __init__(
self,
batchsize_per_replica: int,
shuffle: bool,
transform: Optional[Union[ClassyTransform, Callable]],
num_samples: int,
crop_size: int,
class_ratio: float,
seed: int,
) -> None:
dataset = RandomImageBinaryClassDataset(crop_size, class_ratio,
num_samples, seed,
SampleType.TUPLE)
super().__init__(dataset, batchsize_per_replica, shuffle, transform,
num_samples)