(interp_mode)
| 2249 | |
| 2250 | |
| 2251 | def _to_numpy_resample_interp_mode(interp_mode): |
| 2252 | ret = look_up_option(str(interp_mode), SplineMode, default=None) |
| 2253 | if ret is not None: |
| 2254 | return int(ret) |
| 2255 | _mapping = { |
| 2256 | InterpolateMode.NEAREST: SplineMode.ZERO, |
| 2257 | InterpolateMode.NEAREST_EXACT: SplineMode.ZERO, |
| 2258 | InterpolateMode.LINEAR: SplineMode.ONE, |
| 2259 | InterpolateMode.BILINEAR: SplineMode.ONE, |
| 2260 | InterpolateMode.TRILINEAR: SplineMode.ONE, |
| 2261 | InterpolateMode.BICUBIC: SplineMode.THREE, |
| 2262 | InterpolateMode.AREA: SplineMode.ZERO, |
| 2263 | } |
| 2264 | ret = look_up_option(str(interp_mode), _mapping, default=None) |
| 2265 | if ret is not None: |
| 2266 | return ret |
| 2267 | return look_up_option(str(interp_mode), list(_mapping) + list(SplineMode)) # for better error msg |
| 2268 | |
| 2269 | |
| 2270 | def _to_torch_resample_interp_mode(interp_mode): |
no test coverage detected
searching dependent graphs…