Added network parameter --force_identity_output_resizing to allow regression output shape to match label shape e.g. the case of regressing a single value from image input.
@@ -329,6 +329,13 @@ def add_network_args(parser):
choices=['uniform', 'resize', 'balanced', 'weighted'],
default='uniform')
+ parser.add_argument(
+ "--force_identity_output_resizing",
+ metavar=str2boolean,
+ help="Forces the shape of the inferred output to match the "
+ "input label shape rather than be resized to input image shape.",
+ default=False)
+
parser.add_argument(
"--queue_length",
help="Set size of preprocessing buffer queue",
