迭代字典会抛出 TypeError：列表索引必须是整数或切片，而不是元组

Question

I am trying to iterate over my dictionary with the following code:

num_detections = int(detection['num_detections'])
output_dict = {key:value[0, :num_detections].numpy() 
                for key,value in detection.items()}

But I get

File "main.py", line 60, in for key,value in detection.items()}

TypeError: list indices must be integers or slices, not tuple

When I print out detection it's shaped like a dictionary and print(type(detection)) gives me <class 'dict'>

Am I misunderstanding something?

Answer 1

There is comma in list slicing. I think you wanted this:

num_detections = int(detection['num_detections'])

output_dict = {key:value[0 :num_detections].numpy() 
                for key,value in detection.items()}