Post not yet marked as solved
Post marked as unsolved with 4 replies, 966 views
Hello.I have the follow problem:I have the converted from Caffe(with deconvolutional layer) model which work correct if I run it with python and coremltools. But if I try to load it on Iphone the loader return assertation error:/BuildRoot/Library/Caches/com.apple.xbs/Sources/MetalImage/MetalImage-82.1/MPSNeuralNetwork/Filters/MPSCNNConvolutionTranspose.mm:207: failed assertion `number of groups must be > 0'I now that nGroups correct (=2)Caffe section:layer {
name: "upscore"
type: "Deconvolution"
bottom: "conv10_new"
top: "upscore"
param {
lr_mult: 0
decay_mult: 0
}
convolution_param {
num_output: 2
bias_term: false
pad: 16
kernel_size: 64
group: 2
stride: 32
weight_filler {
type: "bilinear"
}
}
}Mlmodel spec section: layers {
name: "upscore"
input: "conv10_new"
output: "upscore"
convolution {
outputChannels: 2
kernelChannels: 1
nGroups: 2
kernelSize: 64
kernelSize: 64
stride: 32
stride: 32
dilationFactor: 1
dilationFactor: 1
valid {
paddingAmounts {
borderAmounts {
startEdgeSize: 16
endEdgeSize: 16
}
borderAmounts {
startEdgeSize: 16
endEdgeSize: 16
}
}
}
isDeconvolution: true
weights {Any ideas?Thanks!PS. coremltools convert Caffe deconvolution layer not totally correct, it needed to swap outputChannels and kernelChannels (reshape weights in caffe model). For example in this sample caffe model has weights shape (2, 1, 64, 64) but coremltools expect (1, 2, 64, 64)
Posted
by
mikhail_p.
Last updated
mikhail_p.
.