源代码:
f = open("abc.json",'w')
results.append({ "image_id": name, "disease_class":pred.cpu().numpy()[0]})
json_str = json.dumps(results)
f.write(json_str)
原因:json不能处理pred.cpu().numpy()[0] 这种nump 类型数据,
改:
results.append({ "image_id": name, "disease_class":int(pred.cpu().numpy()[0])})