DETR: i/o and keys in output of inference


Sorry if this topic is not appropriate for community.
I develop preprocessing tool kit now;

I use DETR for panopic segmentation (/funcs/ does). The inference’s output have some problem that key (ex. pred_masks) to access makes an error of like a string must not be indexed by integer, or like an error key 0.

I checked source code of the DETR model, it shows used keys, but I could not understand list’s structure. I do not also know expected list structure, so debug is stopped now.

Does someone point out where is possible making such the errors?