diff --git a/model_server/base/roiset.py b/model_server/base/roiset.py index ccd79c2d8f42b59a57e9c143637aed0a5f9a09ef..df3714ceffbeb424d2d10f550c05db26f6109d29 100644 --- a/model_server/base/roiset.py +++ b/model_server/base/roiset.py @@ -19,7 +19,7 @@ from skimage.morphology import binary_dilation, disk from .accessors import GenericImageDataAccessor, InMemoryDataAccessor, write_accessor_data_to_file from .models import InstanceMaskSegmentationModel -from .process import get_safe_contours, pad, rescale, make_rgb, resample_to_8bit +from .process import get_safe_contours, pad, rescale, make_rgb from .annotators import draw_box_on_patch, draw_contours_on_patch, draw_boxes_on_3d_image from .accessors import generate_file_accessor, PatchStack from .process import mask_largest_object @@ -38,6 +38,7 @@ class PatchParams(BaseModel): rgb_overlay_weights: List[float] = [1.0, 1.0, 1.0] pad_to: Union[int, None] = 256 expanded: bool = False + force_tif: bool = False class AnnotatedPatchParams(PatchParams): bounding_box_channel: int = 1