input_config.format = FORMAT_ENUM_TO_INT[input_config.format] if ((input_config.format != mc.ModelInput.FORMAT_NCHW) and (input_config.format != mc.ModelInput.FORMAT_NHWC)): raise Exception("unexpected input format " + mc.ModelInput.Format.Name(input_config.format) + ", expecting " +...
\ --> 101 tensor shape: {}, input_format: {}".format(tensor.shape, input_format) 102 input_format = input_format.upper() 103 AssertionError: size of input tensor and input format are different. tensor shape: (3, 138, 274), input_format: NCHW 解决方案: logger.add_image('images',...
We can see TRT first shuffles NHWC to NCHW (Not sure if this is a bug in visualization) , then reformats it back to NHWC4. We should at least fuse Shuffle and Reformat.spolisetty 2023 年10 月 17 日 17:20 10 Thank you for reporting it. It is missing optimization, and we’ll ...
71 + // NHWC or NCHW format that can be filled with the spatial input features. 72 + float* getBatchEltSpatialInplace(InputBuffers* buffers, int nIdx); 73 + // Returns a pointer to a float array of size getBatchEltGlobalLen() that can be 74 + // filled with the global input...
AssertionError: size of input tensor and input format are different. tensor shape: (3, 138, 274), input_format: NCHW 1. 2. 3. 4. 5. 6. 7. 8. 9. 10. 11. 12. 13. 14. 15. 16. 17. 18. 19. 20. 21. 22. 23. 24. ...