當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python mxnet.ndarray.op.UpSampling用法及代碼示例


用法:

mxnet.ndarray.op.UpSampling(*data, **kwargs)

參數

  • data(NDArray[]) - 要上采樣的張量數組。對於雙線性上采樣,應該有 2 個輸入 - 1 個數據和 1 個權重。
  • scale(int, required) - 上采樣比例
  • num_filter(int, optional, default='0') - 輸入過濾器。僅由雙線性sample_type使用。由於雙線性上采樣使用反卷積,因此將num_filters設置為通道數。
  • sample_type({'bilinear', 'nearest'}, required) - 上采樣方法
  • multi_input_mode({'concat', 'sum'},optional, default='concat') - 如何處理多個輸入。 concat 表示沿通道維度連接上采樣圖像。 sum 表示將所有圖像加在一起,僅適用於最近鄰上采樣。
  • workspace(long (non-negative), optional, default=512) - 反卷積的 Tmp 工作區 (MB)
  • out(NDArray, optional) - 輸出 NDArray 來保存結果。

返回

out- 此函數的輸出。

返回類型

NDArray 或 NDArray 列表

對給定的輸入數據進行上采樣。

兩種算法 (sample_type) 可用於上采樣:

  • 最近的鄰居
  • Bilinear

最近鄰上采樣

輸入數據預計為 NCHW。

例子:

x = [[[[1. 1. 1.]
       [1. 1. 1.]
       [1. 1. 1.]]]]

UpSampling(x, scale=2, sample_type='nearest') = [[[[1. 1. 1. 1. 1. 1.]
                                                   [1. 1. 1. 1. 1. 1.]
                                                   [1. 1. 1. 1. 1. 1.]
                                                   [1. 1. 1. 1. 1. 1.]
                                                   [1. 1. 1. 1. 1. 1.]
                                                   [1. 1. 1. 1. 1. 1.]]]]

雙線性上采樣

在引擎蓋下使用deconvolution 算法。您需要同時提供輸入數據和內核。

輸入數據預計為 NCHW。

num_filter 預計與通道數相同。

例子:

x = [[[[1. 1. 1.]
       [1. 1. 1.]
       [1. 1. 1.]]]]

w = [[[[1. 1. 1. 1.]
       [1. 1. 1. 1.]
       [1. 1. 1. 1.]
       [1. 1. 1. 1.]]]]

UpSampling(x, w, scale=2, sample_type='bilinear', num_filter=1) = [[[[1. 2. 2. 2. 2. 1.]
                                                                     [2. 4. 4. 4. 4. 2.]
                                                                     [2. 4. 4. 4. 4. 2.]
                                                                     [2. 4. 4. 4. 4. 2.]
                                                                     [2. 4. 4. 4. 4. 2.]
                                                                     [1. 2. 2. 2. 2. 1.]]]]

相關用法


注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.op.UpSampling。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。