Skip to content

Commit 7b59931

Browse files
author
Anna Grebneva
authored
Updated output shapes for instance-segmentation models (#3407)
1 parent 0717041 commit 7b59931

File tree

6 files changed

+30
-18
lines changed
  • models/intel

6 files changed

+30
-18
lines changed

models/intel/instance-segmentation-person-0007/README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,14 @@ The expected channel order is `BGR`
3737

3838
## Outputs
3939

40-
1. Name: `labels`, shape: `10` - Contiguous integer class ID for every
40+
Model has outputs with dynamic shapes.
41+
42+
1. Name: `labels`, shape: `-1` - Contiguous integer class ID for every
4143
detected object.
42-
2. Name: `boxes`, shape: `10, 5` - Bounding boxes around every detected objects
44+
2. Name: `boxes`, shape: `-1, 5` - Bounding boxes around every detected objects
4345
in (top_left_x, top_left_y, bottom_right_x, bottom_right_y) format and its
4446
confidence score in range [0, 1].
45-
3. Name: `masks`, shape: `10, 224, 224` - Segmentation heatmaps for every output
47+
3. Name: `masks`, shape: `-1, 224, 224` - Segmentation heatmaps for every output
4648
bounding box.
4749

4850
## Training Pipeline

models/intel/instance-segmentation-security-0002/README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,14 @@ The expected channel order is `BGR`.
3838

3939
## Outputs
4040

41-
1. Name: `labels`, shape: `100` - Contiguous integer class ID for every
41+
Model has outputs with dynamic shapes.
42+
43+
1. Name: `labels`, shape: `-1` - Contiguous integer class ID for every
4244
detected object.
43-
2. Name: `boxes`, shape: `100, 5` - Bounding boxes around every detected objects
45+
2. Name: `boxes`, shape: `-1, 5` - Bounding boxes around every detected objects
4446
in (top_left_x, top_left_y, bottom_right_x, bottom_right_y) format and its
4547
confidence score in range [0, 1].
46-
3. Name: `masks`, shape: `100, 28, 28` - Segmentation heatmaps for every output
48+
3. Name: `masks`, shape: `-1, 28, 28` - Segmentation heatmaps for every output
4749
bounding box.
4850

4951
## Training Pipeline

models/intel/instance-segmentation-security-0091/README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,14 @@ The expected channel order is `BGR`
3838

3939
## Outputs
4040

41-
1. Name: `labels`, shape: `100` - Contiguous integer class ID for every
41+
Model has outputs with dynamic shapes.
42+
43+
1. Name: `labels`, shape: `-1` - Contiguous integer class ID for every
4244
detected object.
43-
2. Name: `boxes`, shape: `100, 5` - Bounding boxes around every detected objects
45+
2. Name: `boxes`, shape: `-1, 5` - Bounding boxes around every detected objects
4446
in (top_left_x, top_left_y, bottom_right_x, bottom_right_y) format and its
4547
confidence score in range [0, 1].
46-
3. Name: `masks`, shape: `100, 28, 28]` - Segmentation heatmaps for every output
48+
3. Name: `masks`, shape: `-1, 28, 28` - Segmentation heatmaps for every output
4749
bounding box.
4850

4951
## Training Pipeline

models/intel/instance-segmentation-security-0228/README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,14 @@ The expected channel order is `BGR`
3636

3737
## Outputs
3838

39-
1. Name: `labels`, shape: `100` - Contiguous integer class ID for every
39+
Model has outputs with dynamic shapes.
40+
41+
1. Name: `labels`, shape: `-1` - Contiguous integer class ID for every
4042
detected object.
41-
2. Name: `boxes`, shape: `100, 5` - Bounding boxes around every detected objects
43+
2. Name: `boxes`, shape: `-1, 5` - Bounding boxes around every detected objects
4244
in (top_left_x, top_left_y, bottom_right_x, bottom_right_y) format and its
4345
confidence score in range [0, 1].
44-
3. Name: `masks`, shape: `100, 14, 14` - Segmentation heatmaps for every output
46+
3. Name: `masks`, shape: `-1, 14, 14` - Segmentation heatmaps for every output
4547
bounding box.
4648

4749
## Training Pipeline

models/intel/instance-segmentation-security-1039/README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,14 @@ The expected channel order is `BGR`
3636

3737
## Outputs
3838

39-
1. Name: `labels`, shape: `100` - Contiguous integer class ID for every
39+
Model has outputs with dynamic shapes.
40+
41+
1. Name: `labels`, shape: `-1` - Contiguous integer class ID for every
4042
detected object.
41-
2. Name: `boxes`, shape: `100, 5` - Bounding boxes around every detected objects
43+
2. Name: `boxes`, shape: `-1, 5` - Bounding boxes around every detected objects
4244
in (top_left_x, top_left_y, bottom_right_x, bottom_right_y) format and its
4345
confidence score in range [0, 1].
44-
3. Name: `masks`, shape: `100, 14, 14` - Segmentation heatmaps for every output
46+
3. Name: `masks`, shape: `-1, 14, 14` - Segmentation heatmaps for every output
4547
bounding box.
4648

4749
## Training Pipeline

models/intel/instance-segmentation-security-1040/README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,14 @@ The expected channel order is `BGR`
3636

3737
## Outputs
3838

39-
1. Name: `labels`, shape: `100` - Contiguous integer class ID for every
39+
Model has outputs with dynamic shapes.
40+
41+
1. Name: `labels`, shape: `-1` - Contiguous integer class ID for every
4042
detected object.
41-
2. Name: `boxes`, shape: `100, 5` - Bounding boxes around every detected objects
43+
2. Name: `boxes`, shape: `-1, 5` - Bounding boxes around every detected objects
4244
in (top_left_x, top_left_y, bottom_right_x, bottom_right_y) format and its
4345
confidence score in range [0, 1].
44-
3. Name: `masks`, shape: `100, 28, 28` - Segmentation heatmaps for every output
46+
3. Name: `masks`, shape: `-1, 28, 28` - Segmentation heatmaps for every output
4547
bounding box.
4648

4749
## Training Pipeline

0 commit comments

Comments
 (0)