def to_proto(self): msg = super().to_proto() chip_options = TaskConfigMsg.SemanticSegmentationConfig.ChipOptions( window_method=self.chip_options.window_method, target_classes=self.chip_options.target_classes, debug_chip_probability=self.chip_options.debug_chip_probability, negative_survival_probability=self.chip_options. negative_survival_probability, chips_per_scene=self.chip_options.chips_per_scene, target_count_threshold=self.chip_options.target_count_threshold, stride=self.chip_options.stride) conf = TaskConfigMsg.SemanticSegmentationConfig( chip_size=self.chip_size, class_items=self.class_map.to_proto(), chip_options=chip_options) msg.MergeFrom( TaskConfigMsg(semantic_segmentation_config=conf, predict_package_uri=self.predict_package_uri)) return msg
def to_proto(self): msg = super().to_proto() chip_options = TaskConfigMsg.ObjectDetectionConfig.ChipOptions( neg_ratio=self.chip_options.neg_ratio, ioa_thresh=self.chip_options.ioa_thresh, window_method=self.chip_options.window_method, label_buffer=self.chip_options.label_buffer) predict_options = TaskConfigMsg.ObjectDetectionConfig.PredictOptions( merge_thresh=self.predict_options.merge_thresh, score_thresh=self.predict_options.score_thresh) conf = TaskConfigMsg.ObjectDetectionConfig( chip_size=self.chip_size, class_items=self.class_map.to_proto(), chip_options=chip_options, predict_options=predict_options) msg.MergeFrom( TaskConfigMsg(object_detection_config=conf, predict_package_uri=self.predict_package_uri)) return msg
def to_proto(self): conf = TaskConfigMsg.ChipClassificationConfig( chip_size=self.chip_size, class_items=self.class_map.to_proto()) return TaskConfigMsg(task_type=rv.CHIP_CLASSIFICATION, chip_classification_config=conf, predict_package_uri=self.predict_package_uri)