zlszhonglongshen / pytorch-unet-1

pytorch-unet

• 来源于usuyama/pytorch-unet
• 训练字幕图像的背景擦除模型

目录结构

.
├── convert.py
├── datasets
│   └── gen_datasets
│            ├── train
│            │   ├── images
│            │   └── masks
│            └── val
│                ├── images
|                └── masks
├── eval_onnxruntime.py
├── loss.py
├── pytorch_unet.py
├── requirements.txt
└── train_self.py