将Pytorch的图像格式从NHWC更改为NCHW

时间:2018-08-16 16:20:08

标签: image deep-learning permutation conv-neural-network pytorch

在pytorch中,我们需要public class IgnoreTestListener implements IInvokedMethodListener { @Override public void beforeInvocation(IInvokedMethod iInvokedMethod, ITestResult testResult) { Method method = iInvokedMethod.getTestMethod().getConstructorOrMethod().getMethod(); if (method.isAnnotationPresent(EnvDependentIgnore.class)) { if (//"your environment value" .equalsIgnoreCase(method.getAnnotation(EnvDependentIgnore.class).env().getValue()) && method.getAnnotation(EnvDependentIgnore.class).skip()) { throw new SkipException(method.getAnnotation(EnvDependentIgnore.class).reason()); } } } @Override public void afterInvocation(IInvokedMethod method, ITestResult testResult) { // no need to implement } } 格式的图像,但我的图像是NCHW

将该图像馈送到CNN的过程是什么? (我发现this solution建议使用“置换”,但我应该在哪里以及如何使用它?)

1 个答案:

答案 0 :(得分:3)

使用torch.Tensor.permute()

 x = x.permute(0, 3, 1, 2) # from NHWC to NCHW