KinD算法运行过程

本文档详细介绍了如何运行KinD算法。首先,通过GIT bash下载并覆盖预训练的检查点。接着,在TensorFlow虚拟环境中运行代码时遇到了库未安装(如scikit-image)和GPU内存不足的问题。解决这些问题后,算法成功运行,对KinD est目录中的图片进行增强,并将结果保存在KinD esult est。最后,将数据集替换为Dark face,比较了与EnlightenGAN的增强效果。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

源码:https://github.com/zhangyhuaee/KinD

1.打开GIT bash

git clone https://github.com/zhangyhuaee/KinD.git

2.下载预训练的检查点checkpoint,解压后与源代码相比多了一个文件,我选择直接覆盖了原文件中的checkpoint。

3.在刚刚创建的配置了TensorFlow的虚拟环境中,运行代码:

python evaluate.py

出现很多问题:

一:很多库没安装:(skimagea安装报错,因为他本命叫scikit-image)

conda install scipy
conda install scikit-image

二:InternalError:failed to create session

GPU内存不够了,在代码前设置export CUDA_VISIBLE_DEVICE=0。

问题解决后:

查看路径,可以看到是对KinD\test中的三张图片进行增强,结果保存在KinD\result\test中:

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值