超分辨率重建数据集制作:高分辨率图像采集(从pdf中获取图像)
目录
背景
代码
其他
背景
超分辨率重建任务首先需要高清图像。由于需要自己制作超分辨率重建数据集,我从pdf中获取我想要得图像,我根据网上代码进行了改造
可运行代码如下:(注意修改最后两行中的pdf路径和保存结果路径)
代码
#coding=utf-8
import os
import fitz
from os.path import abspath, dirname
# 使用fitz 库直接提取pdf的图像
# 参数: pdf 源pdf文件完整路径
# 参数: picPath 提取图像的路径
def muExtractImages(pdf, picPath):
pdfsplit = os.path.split(pdf) # 分离出文件名和路径
pdfname = pdfsplit[-1] # 获取文件名
pdfsplit1 = os.path.splitext(pdfname)
pdfname1 = pdfsplit1[0] # 获取不带扩展名的文件名
共有 0 条评论