def process_arguments(self, arguments): self.arguments = arguments print("Input Directory: {}".format(self.arguments.input_dir)) print("Output Directory: {}".format(self.arguments.output_dir)) self.serializer = None if self.arguments.serializer is None and self.arguments.alignments_path is not None: ext = os.path.splitext(self.arguments.alignments_path)[-1] self.serializer = Serializer.get_serializer_fromext(ext) print(self.serializer, self.arguments.alignments_path) else: self.serializer = Serializer.get_serializer(self.arguments.serializer or "json") print("Using {} serializer".format(self.serializer.ext)) print('Starting, this may take a while...') self.output_dir = get_folder(self.arguments.output_dir) try: self.input_dir = get_image_paths(self.arguments.input_dir) except: print('Input directory not found. Please ensure it exists.') exit(1) self.filter = self.load_filter() self.process() self.finalize()
def get_serializer(self): """ Set the serializer to be used for loading and saving alignments """ if not self.args.serializer and self.args.alignments_path: ext = os.path.splitext(self.args.alignments_path)[-1] serializer = Serializer.get_serializer_fromext(ext) print("Alignments Output: {}".format(self.args.alignments_path)) else: serializer = Serializer.get_serializer(self.args.serializer or "json") print("Using {} serializer".format(serializer.ext)) return serializer
def process_arguments(self, arguments): self.arguments = arguments print("Input Directory: {}".format(self.arguments.input_dir)) print("Output Directory: {}".format(self.arguments.output_dir)) print("Filter: {}".format(self.arguments.filter)) self.serializer = None if self.arguments.serializer is None and self.arguments.alignments_path is not None: ext = os.path.splitext(self.arguments.alignments_path)[-1] self.serializer = Serializer.get_serializer_fromext(ext) print(self.serializer, self.arguments.alignments_path) else: self.serializer = Serializer.get_serializer( self.arguments.serializer or "json") print("Using {} serializer".format(self.serializer.ext)) try: if self.arguments.rotate_images is not None and self.arguments.rotate_images != "off": if self.arguments.rotate_images == "on": self.rotation_angles = range(90, 360, 90) else: rotation_angles = [ int(angle) for angle in self.arguments.rotate_images.split(",") ] if len(rotation_angles) == 1: rotation_step_size = rotation_angles[0] self.rotation_angles = range(rotation_step_size, 360, rotation_step_size) elif len(rotation_angles) > 1: self.rotation_angles = rotation_angles except AttributeError: pass print('Starting, this may take a while...') try: if self.arguments.skip_existing: self.already_processed = get_image_paths( self.arguments.output_dir) except AttributeError: pass self.output_dir = get_folder(self.arguments.output_dir) try: try: if self.arguments.skip_existing: self.input_dir = get_image_paths(self.arguments.input_dir, self.already_processed) print('Excluding %s files' % len(self.already_processed)) else: self.input_dir = get_image_paths(self.arguments.input_dir) except AttributeError: self.input_dir = get_image_paths(self.arguments.input_dir) except: print('Input directory not found. Please ensure it exists.') exit(1) self.filter = self.load_filter() self.process() self.finalize()