一、数据集加载
1. 数据集放置
将下载的数据集解压到datasets文件夹的下面,方便后续操作。
2. 数据集预处理
数据集预处理针对两个数据集给了两个不同的预处理指令,我测试了VISA数据集,以下是我对VISA数据集的实例。
1. datasets/prepare_visa_public.py文件配置
打开这个文件,第16行代码后有很多文件地址的设置,default后面就跟着一个地址,这个地址是“../datasets/xxxx/”我把这几行注释掉,然后改为:“./datasets/xxxxx/”,这样就是返回上一级目录,而不是两级。(如图所示)
2. datasets/prepare_visa_public.py文件运行
运行此文件我是在命令行进行的,点开terminal终端,输入
python datasets/prepare_visa_public.py
运行成功后就会在datasets中产生一个新的数据包“datasets\VisA_pytorch”
二、 测试模型结果
直接在终端里输入
python run_winclip.py
然后就会有结果产生啦~~~
其中做的时候会遇到一些报错,最常遇到的是loguru库的安装失败
如果出现了相关报错可以参考这篇:【python】之loguru库,好用的日志管理库!_loguru安装-CSDN博客