Searched refs:backbone_shape (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | yolo.py | 121 def __init__(self, backbone_shape, backbone, out_channel): argument
125 …self.backblock0 = YoloBlock(backbone_shape[-1], out_chls=backbone_shape[-2], out_channels=out_chan…
127 …self.conv1 = _conv_bn_relu(in_channel=backbone_shape[-2], out_channel=backbone_shape[-2]//2, ksize…
128 self.backblock1 = YoloBlock(in_channels=backbone_shape[-2]+backbone_shape[-3],
129 out_chls=backbone_shape[-3],
132 …self.conv2 = _conv_bn_relu(in_channel=backbone_shape[-3], out_channel=backbone_shape[-3]//2, ksize…
133 self.backblock2 = YoloBlock(in_channels=backbone_shape[-3]+backbone_shape[-4],
134 out_chls=backbone_shape[-4],
372 self.config.backbone_shape,
374 backbone_shape=self.config.backbone_shape,
|
| D | config.py | 48 backbone_shape = [64, 128, 256, 512, 1024] variable in ConfigYOLOV3DarkNet53
|
| /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
343 …self.backblock0 = YoloBlock(backbone_shape[-1], out_chls=backbone_shape[-2], out_channels=out_chan…
345 …self.conv1 = _conv_bn_relu(in_channel=backbone_shape[-2], out_channel=backbone_shape[-2]//2, ksize…
347 self.backblock1 = YoloBlock(in_channels=backbone_shape[-2]+backbone_shape[-3],
348 out_chls=backbone_shape[-3],
351 …self.conv2 = _conv_bn_relu(in_channel=backbone_shape[-3], out_channel=backbone_shape[-3]//2, ksize…
353 self.backblock2 = YoloBlock(in_channels=backbone_shape[-3]+backbone_shape[-4],
354 out_chls=backbone_shape[-4],
587 self.config.backbone_shape,
590 backbone_shape=self.config.backbone_shape,
|
| D | config.py | 32 backbone_shape = [64, 128, 256, 512] variable in ConfigYOLOV3ResNet18
|