# 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
- 粉丝: 3129
- 资源: 754
最新资源
- MBR污水一体化处理系统(集装箱)工程图机械结构设计图纸和其它技术资料和技术方案非常好100%好用.zip
- lattice diamond3.11的license文件
- vsvbxcygsyzgvytfvdyvs
- DJS-042-锁螺丝机方案工程图机械结构设计图纸和其它技术资料和技术方案非常好100%好用.zip
- FMASTERSW3.2版本
- qemu上运行Linux系统开启并验证IMA功能
- HE-Drive-main.zip
- mysql安装配置教程.txt
- mysql安装配置教程.txt
- mysql安装配置教程.txt
- 汇川伺服6V30-EOE-MXL文件-037515
- 2024中国CIO&CDO现状、挑战及未来趋势研究报告
- 鼠标连点器+自动单机+录制点击
- 4G DTU串口数据采集网关设计全套资料(源码、原理图、外壳文件、产品手册).zip
- 利用Python绘制装饰圣诞树的技术实例
- 测试程序:qabstractvideosurface
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈