使用mxnet.ndarray.UpSampling实现双线性插值

作者: star
发布时间: 2024-11-13 06:36:52 (3天前)
转自：

2 条回复

0#
回复此人
电动少女 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> <code> mxnet.ndarray.UpSampling </code> 似乎期望双线性sample_type有2个输入（1输入和1个权重） </p> <P> 此外，我认为缺少num_args参数的文档，您可以在此处查看。 <a href="https://github.com/apache/incubator-mxnet/blob/master/src/operator/nn/upsampling-inl.h#L78" rel="nofollow noreferrer"> https://github.com/apache/incubator-mxnet/blob/master/src/operator/nn/upsampling-inl.h#L78 </A> </p> <P> 这应该工作： </p> <pre> <code> import mxnet as mx import mxnet.ndarray as nd xx = nd.random_normal(shape=[1,1,256,256],ctx=mx.cpu()) xx1 = nd.random_normal(shape=[1,1,4,4],ctx=mx.cpu()) temp = nd.UpSampling(xx,xx1, num_filter=1, scale=2, sample_type='bilinear', num_args=2) </code> </pre> </DIV>

编辑

登录后才能参与评论