def setUp(self):
super().setUp()
assert replace_text_in_file(self.configuration_file,
'data/WIDERFace', '../../data/airport')
assert replace_text_in_file(self.configuration_file, 'val.json',
'annotation_faces_train.json')
def setUp(self):
self.model_name = 'person-vehicle-bike-detection-crossroad-1016'
self.snapshot_name = 'person_vehicle_bike_sd512_mb2_clustered_epoch_21.pth'
self.data_folder = '../../data'
self.work_dir = os.path.join('/tmp', self.model_name)
os.makedirs(self.work_dir, exist_ok=True)
self.configuration_file = f'./configs/{self.model_name}.py'
os.system(f'cp {self.configuration_file} {self.work_dir}/')
self.configuration_file = os.path.join(
self.work_dir, os.path.basename(self.configuration_file))
self.ote_url = 'https://download.01.org/opencv/openvino_training_extensions'
self.url = f'{self.ote_url}/models/object_detection/{self.snapshot_name}'
download_if_not_yet(self.work_dir, self.url)
assert replace_text_in_file(self.configuration_file, 'total_epochs =',
'total_epochs = 25#')
assert replace_text_in_file(
self.configuration_file, 'work_dir =',
f'work_dir = "{os.path.join(self.work_dir, "outputs")}" #')
assert replace_text_in_file(
self.configuration_file, 'load_from = None',
f'load_from = "{os.path.join(self.work_dir, self.snapshot_name)}"')
assert replace_text_in_file(self.configuration_file,
'annotation_example_train.json',
'annotation_example_val.json')
assert replace_text_in_file(self.configuration_file, '/train', '/val')
def setUp(self):
super().setUp()
assert replace_text_in_file(
self.configuration_file, 'data/text-dataset/',
'../../data/horizontal_text_detection/')
assert replace_text_in_file(self.configuration_file,
'IC13TEST.json', 'annotation.json')
def setUp(self):
self.model_name = model_name
self.data_folder = '../../data'
self.work_dir = os.path.join('/tmp/', self.model_name)
os.makedirs(self.work_dir, exist_ok=True)
self.configuration_file = f'./face-detection/{self.model_name}/config.py'
run_through_shell(f'cp {self.configuration_file} {self.work_dir}/')
self.configuration_file = os.path.join(
self.work_dir, os.path.basename(self.configuration_file))
self.ote_url = 'https://download.01.org/opencv/openvino_training_extensions'
self.url = f'{self.ote_url}/models/object_detection/v2/{self.model_name}.pth'
download_if_not_yet(self.work_dir, self.url)
assert replace_text_in_file(self.configuration_file,
'samples_per_gpu=',
'samples_per_gpu=1 ,#')
assert replace_text_in_file(self.configuration_file,
'total_epochs = 70',
'total_epochs = 75')
assert replace_text_in_file(self.configuration_file,
'data/WIDERFace', '../../data/airport')
assert replace_text_in_file(
self.configuration_file, 'work_dir =',
f'work_dir = "{os.path.join(self.work_dir, "outputs")}" #')
assert replace_text_in_file(self.configuration_file, 'train.json',
'annotation_faces_train.json')
assert replace_text_in_file(self.configuration_file, 'val.json',
'annotation_faces_train.json')
assert replace_text_in_file(
self.configuration_file, 'resume_from = None',
f'resume_from = "{os.path.join(self.work_dir, self.model_name)}.pth"'
)
def setUp(self):
self.model_name = model_name
self.data_folder = '../../data/horizontal_text_detection'
self.work_dir = os.path.join('/tmp', self.model_name)
os.makedirs(self.work_dir, exist_ok=True)
self.configuration_file = f'./horizontal-text-detection/{self.model_name}/config.py'
run_through_shell(f'cp {self.configuration_file} {self.work_dir}/')
self.configuration_file = os.path.join(
self.work_dir, os.path.basename(self.configuration_file))
self.ote_url = 'https://download.01.org/opencv/openvino_training_extensions'
self.url = f'{self.ote_url}/models/object_detection/v2/{self.model_name}.pth'
download_if_not_yet(self.work_dir, self.url)
assert replace_text_in_file(self.configuration_file,
'total_epochs = 25',
'total_epochs = 35')
assert replace_text_in_file(
self.configuration_file, 'work_dir =',
f'work_dir = "{os.path.join(self.work_dir, "outputs")}" #')
assert replace_text_in_file(
self.configuration_file, 'resume_from = None',
f'resume_from = "{os.path.join(self.work_dir, self.model_name)}.pth"'
)
assert replace_text_in_file(
self.configuration_file, 'data/text-dataset/',
'../../data/horizontal_text_detection/')
assert replace_text_in_file(
self.configuration_file,
'IC13TRAIN_IC15_IC17_IC19_MSRATD500_COCOTEXT.json',
'annotation.json')
assert replace_text_in_file(self.configuration_file,
'IC13TEST.json', 'annotation.json')