python 列表保存爲json文件錯誤: TypeError: 2 is not JSON serializable

源代碼:

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])})

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章