| 276 | } |
| 277 | |
| 278 | int64_t GetOutputHeight(int64_t height, NVCVColorConversionCode code) |
| 279 | { |
| 280 | switch (code) |
| 281 | { |
| 282 | case NVCVColorConversionCode::NVCV_COLOR_YUV2RGB_NV12: |
| 283 | case NVCVColorConversionCode::NVCV_COLOR_YUV2BGR_NV12: |
| 284 | case NVCVColorConversionCode::NVCV_COLOR_YUV2RGB_NV21: |
| 285 | case NVCVColorConversionCode::NVCV_COLOR_YUV2BGR_NV21: |
| 286 | return (2 * height) / 3; // output height must be 2/3 of input height from NV12 or NV21 |
| 287 | |
| 288 | case NVCVColorConversionCode::NVCV_COLOR_RGB2YUV_NV12: |
| 289 | case NVCVColorConversionCode::NVCV_COLOR_BGR2YUV_NV12: |
| 290 | case NVCVColorConversionCode::NVCV_COLOR_RGB2YUV_NV21: |
| 291 | case NVCVColorConversionCode::NVCV_COLOR_BGR2YUV_NV21: |
| 292 | return (3 * height) / 2; // output height must be 3/2 of input height for UV plane |
| 293 | |
| 294 | default: |
| 295 | return height; |
| 296 | } |
| 297 | } |
| 298 | |
| 299 | nvcv::TensorShape GetOutputTensorShape(const nvcv::TensorShape &inputShape, nvcv::ImageFormat outputFormat, |
| 300 | NVCVColorConversionCode code) |
no outgoing calls
no test coverage detected