|
发表于 2024-4-23 18:05:44
|
显示全部楼层
感谢大佬,希望可以早日将新版分享到网盘链接。
现在网盘链接里最新的2.2.0的保存loss到json部分的代码缩进有点问题,会导致保存loss时出错:
if samples_tuple[0] != 0:
# 获取样本文件名的.stem
src_file_name = Path(samples_tuple[0].filename).stem
# 将处理后的图像保存为jpg文件
cv2_imwrite(
self.src_state_path / f"{src_file_name}_output.jpg",
self._get_formatted_image(pred_src_src) * 255,
[int(cv2.IMWRITE_JPEG_QUALITY), 100],
) # output
src_data = {
# 将src_loss的第一个元素转换为浮点数并赋值给loss键
"loss": float(src_loss[0]),
# 将src_file_name加上后缀.jpg并赋值给input键
"input": f"{src_file_name}.jpg",
# 将src_file_name加上后缀_output.jpg并赋值给output键
"output": f"{src_file_name}_output.jpg",
}
# 将src_data添加到self.src_sample_state列表中
self.src_sample_state.append(src_data)
这是我手动改了下缩进,就可以正常使用了。 |
|