File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
rastervision_pytorch_backend/rastervision/pytorch_backend/examples/semantic_segmentation Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -44,6 +44,7 @@ def get_config(runner,
44
44
external_model : bool = True ,
45
45
augment : bool = False ,
46
46
nochip : bool = True ,
47
+ allow_streaming : bool = False ,
47
48
num_epochs : int = 10 ,
48
49
batch_sz : int = 8 ,
49
50
test : bool = False ) -> SemanticSegmentationConfig :
@@ -142,7 +143,9 @@ def make_scene(id) -> SceneConfig:
142
143
label_uri = label_crop_uri
143
144
144
145
raster_source = RasterioSourceConfig (
145
- uris = [raster_uri ], channel_order = channel_order )
146
+ uris = [raster_uri ],
147
+ channel_order = channel_order ,
148
+ allow_streaming = allow_streaming )
146
149
147
150
# Using with_rgb_class_map because label TIFFs have classes encoded as
148
151
# RGB colors.
@@ -151,7 +154,8 @@ def make_scene(id) -> SceneConfig:
151
154
uris = [label_uri ],
152
155
transformers = [
153
156
RGBClassTransformerConfig (class_config = class_config )
154
- ]))
157
+ ],
158
+ allow_streaming = allow_streaming ))
155
159
156
160
# URI will be injected by scene config.
157
161
# Using rgb=True because we want prediction TIFFs to be in
You can’t perform that action at this time.
0 commit comments