image_pred = image_pred[(-score).argsort()]
yolov3程序調試時出錯:pytorch的tensor沒有argsort方法
不希望轉成numpy再轉回tensor,查閱手冊發現 pytorch的sort同時返回了sorted和indices,代碼修改爲:
srtd, indices = torch.sort(-score)
image_pred = image_pred[indices]
附上numpy和pytorch對照表,來自pytorch中文網:
面向 Numpy 用戶的 PyTorch 速查表