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)