def setUp(self):
self.image_data = test_utils.get_test_df()
self.split_key = schema.get_key(schema.SplitKeyType,
schema.image_csv_schema)
self.label_key = schema.get_key(schema.StringLabelType,
schema.image_csv_schema)
self.image_uri_key = schema.get_key(schema.ImageUriType,
schema.image_csv_schema)
self.tempfiles = []
self.tempdir = None
def setUp(self):
self.test_df = test_utils.get_test_df()
self.test_region = 'us-central1'
self.test_project = 'foo'
self.test_wheel = '/my/path/wheel.whl'
def setUp(self):
self.test_df = test_utils.get_test_df()
self.test_region = 'us-central1'
self.test_project = 'foo'
self.test_wheel = '/my/path/wheel.whl'
self.test_schema_map = schema.image_csv_schema
def setUp(self): self.df = test_utils.get_test_df() self.schema_map = schema.image_csv_schema self.split_key = schema.get_key(schema.SplitKeyType, self.schema_map)
def setUp(self):
self.test_df = test_utils.get_test_df()
self.test_region = 'us-central1'
self.test_project = 'foo'
self.test_wheel = '/my/path/wheel.whl'
self.test_schema = input_schema.IMAGE_CSV_SCHEMA
def setUp(self):
self.image_data = test_utils.get_test_df()
self.tempfiles = []
self.tempdir = None
self.schema = input_schema.Schema(
input_schema.IMAGE_CSV_SCHEMA.input_schema_map)
def setUp(self):
self.pipeline = test_utils.get_test_pipeline()
self.df = test_utils.get_test_df()
self.image_file = 'tfrecorder/test_data/images/cat/cat-640x853-1.jpg'
def setUp(self):
self.df = test_utils.get_test_df()
self.schema = input_schema.IMAGE_CSV_SCHEMA
self.schema_map = self.schema.input_schema_map
self.split_key = self.schema.split_key
def setUp(self):
self.test_df = test_utils.get_test_df()
self.test_region = 'us-central1'
self.test_project = 'foo'
def setUp(self):
self.test_df = test_utils.get_test_df()
self.output_dir = '/tmp/train'
os.makedirs(self.output_dir, exist_ok=True)
def setUp(self):
self.pipeline = test_utils.get_test_pipeline()
self.df = test_utils.get_test_df()
self.image_file = self.df.image_uri.sample(
random_state=RANDOM_SEED).values[0]