[Tag NCS_Beta_v1.3] In "training/data_loader/data_loader.py", `readData` called by `getPatchData` raises an `ValueError` for some values of the second argument `comp` of `getPatchData`.

Problem

If, in the script provided in the README.txt at Section "Data loading", comps is ['qp_base', 'pred_U', 'pred_V'], and this script is run, readData called by getPatchData raises an ValueError at line https://vcgit.hhi.fraunhofer.de/jvet-ahg-nnvc/VVCSoftware_VTM/-/blob/NCS_Beta_v1.3/training/data_loader/data_loader.py#L18.

As another example, let us say that the following command is run:

$ python training/example/create_unified_dataset.py --input_json dataset.json --components qp_base,pred_U,pred_V --output_file file_final.pb

Then, the same kind of exception is raised by readData called by getPatchData in "training/data_loader/data_loader.py".

Explanation

At this line https://vcgit.hhi.fraunhofer.de/jvet-ahg-nnvc/VVCSoftware_VTM/-/blob/NCS_Beta_v1.3/training/data_loader/data_loader.py#L121, getPatchData identifies whether the data to be loaded is Y or U/V according to the first string in the second argument comp of getPatchData. If the first string in the second argument comp is 'qp_base', the data to be loaded is identified as Y whereas the next data to be loaded (characterized by 'pred_U' and 'pred_V') are actually U/V. The reader thus uses wrong frame dimensions.