显而易见,是pytorch中的Tensor和numpy.ndarry类型导致的问题,只需要进行array和tensor的转换就行
调用函数torch.from_numpy(annotation)实现array转换tensor