视频字幕识别（百度AI开放平台OCR

CÃlimos: 请问是这样改吗 def centernet_dataset_collate(batch): imgs, batch_hms, batch_box, batch_regs, batch_reg_masks= [], [], [], [], [] for img, batch_hm, batch_box, batch_reg, batch_reg_mask in batch: imgs.append(img) batch_hms.append(batch_hm) batch_box.append(batch_box)#这里报错：AttributeError: 'numpy.ndarray' object has no attribute 'append' batch_regs.append(batch_reg) batch_reg_masks.append(batch_reg_mask) batch_box=np.array(batch_box)#这里转换一下 imgs = torch.from_numpy(np.array(imgs)).type(torch.FloatTensor) batch_hms = torch.from_numpy(np.array(batch_hms)).type(torch.FloatTensor) batch_box =torch.from_numpy(np.array(batch_box)).type(torch.FloatTensor) batch_regs = torch.from_numpy(np.array(batch_regs)).type(torch.FloatTensor) batch_reg_masks = torch.from_numpy(np.array(batch_reg_masks)).type(torch.FloatTensor) return imgs, batch_hms,batch_box, batch_regs, batch_reg_masks 但还是报错

   ゞ 草莓布丁: 还是不知道怎么解决