Searched refs:feature_shape (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/st/networks/models/deeplabv3/src/ |
| D | deeplabv3.py | 26 def __init__(self, feature_shape, scale_size, output_stride): argument
28 sample_h = (feature_shape[0] * scale_size + 1) / output_stride + 1
29 sample_w = (feature_shape[1] * scale_size + 1) / output_stride + 1
55 def __init__(self, channel, depth, feature_shape, scale_sizes, argument
84 aspp_scale_depth_size = np.ceil((feature_shape[0]*scale_size)/16)
105 … self.global_pooling = nn.AvgPool2d(kernel_size=(int(feature_shape[0]), int(feature_shape[1])))
108 pooling_h = np.ceil((feature_shape[0]*scale_size)/output_stride)
109 pooling_w = np.ceil((feature_shape[0]*scale_size)/output_stride)
119 self.samples.append(ASPPSampleBlock(feature_shape, scale_size, output_stride))
121 self.feature_shape = feature_shape
[all …]
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | yolov3.py | 339 def __init__(self, feature_shape, backbone_shape, backbone, out_channel): argument
346 self.upsample1 = P.ResizeNearestNeighbor((feature_shape[2]//16, feature_shape[3]//16))
352 self.upsample2 = P.ResizeNearestNeighbor((feature_shape[2]//8, feature_shape[3]//8))
583 self.feature_map = YOLOv3(feature_shape=self.config.feature_shape,
|
| D | config.py | 27 feature_shape = [32, 3, 352, 640] variable in ConfigYOLOV3ResNet18
|