# UNet Seal Elimination
UNet 实现文档印章消除
# Requirement
pytorch==1.5
opencv-python 4.2
numpy
# Use
1. data 的目录结构如下图:
![data目录结构](./images/data.png)
- test:测试集图片的路径
- mytest:测试结果的输出路径
- train:训练集图片的路径,包含含印章的图片以及标注印章位置的xml文件
- train_cleaned:训练集图片人工去除印章后的标签
- valid:验证集图片的路径,包含含印章的图片以及标注印章位置的xml文件
- valid_cleaned:验证集图片人工去除印章后的标签
2. config.py
设置参数,包括文件路径、模型结构参数和训练的参数等。
3. train.py
运行 python train.py 训练模型。
4. predict.py
运行 python predict.py 测试。
# Note:
1. 由于作者所使用的图像分辨率极高,在训练和测试时从完整图像中扣出包含印章的区域(ImageSize=512*512),然后进行训练。如果图片的分辨率适中或者显存足够大,可以跳过此步骤,无需进行印章标注,直接使用原图进行UNet训练。
2. 从原图中扣出印章区域也可以使用yolo代替。
DdddJMs__135
- 粉丝: 3072
- 资源: 727
最新资源
- Ruby - Ruby 开发 - 常用知识点
- 响应式营销型运动健身器材pbootcms网站模板
- ingress.yaml
- LabVIEW练习44,计算学生三门课(语文,数学,英语)的平均分,并根据平均分划分成绩等级
- densenet模型-基于深度学习对时尚配饰识别-不含数据集图片-含逐行注释和说明文档.zip
- 【C语音期末/课程设计】银行客户管理系统(DevC项目)
- densenet模型-基于深度学习识别电子产品-不含数据集图片-含逐行注释和说明文档.zip
- shufflenet模型-基于卷积神经网络识别地理特征-不含数据集图片-含逐行注释和说明文档.zip
- 西北工业大学编译原理试点班大作业-实现一个能够正常工作的Sysy语法编译器+源代码+文档说明+实验报告
- shufflenet模型-图像分类算法对农作物种类识别-不含数据集图片-含逐行注释和说明文档.zip
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈