I was able to solve the 2nd issue related to a key ‘image_id’ not available. It was my error and I noticed after double reading the example here 13. I only added the ‘boxes’ and ‘labels’ keys to the target dictionary but the keys ‘image_id’, ‘area’ and ‘iscrowd’ are also necessary. I created them by following the same example.
Any help on the CUDA Out of memory error will be welcomed.