# DTFED
## Introduction
DTFED: 面向机器学习的纹理滤波与边缘检测训练数据集
> 本数据集可于纹理滤波和边缘检测任务的训练
## Requirements
- Python 3.7
- pytorch 1.8.1
- matplotlib 3.4.1
## Usage
### Installation
- Download the repository.
```
git clone https://gitee.com/Chinazjk/DTFED.git
```
- Install python dependencies.
```
pip3 install torch torchvision torchaudio
pip3 install matplotlib
```
- Download the dataset.
> [百度云盘](https://pan.baidu.com/s/1-AriDUY8-m-LaFCrCYTrqg) 提取码: ccik
### Test
#### BDCN
> run `test_DTFED.py` in `.\BDCN`
>```
># 如果需要测试BSDS500、NYUD数据集训练的模型
>run test_DTFED.py -t [BSDS500\NYUD]
>```
#### BEPS
> run `BEPS_train_by_TED.py` in `.\BEPS`
>
> 由于版本原因，本文复现使用自己的代码模型结构和官方相同,使用pytorch复现。运行`BEPS_train_by_TED.py`即可运行，结果存储在`BEPS_train_byTED`中 我们提供了重新训练的模型`model45_2`
而运行官方实现的代码通过运行`VDCNN_tf/test.py` 结果会保存在`VDCNN_tf/result`中。
>
>*note*: 官方是使用pytorch实现的
### Result
> 结果分别保存在对应方法的 `result` 目录下
面向机器学习的纹理滤波与边缘检测训练数据集，本数据集可于纹理滤波和边缘检测任务的训练（300M） （125个子文件） 
20.bmp 1.41MB 20.bmp 1.41MB 37.bmp 929KB 37.bmp 929KB 24.bmp 792KB 24.bmp 792KB 91.bmp 503KB 91.bmp 503KB 20.bmp 484KB 20.bmp 484KB 37.bmp 312KB 37.bmp 312KB 24.bmp 266KB 24.bmp 266KB 91.bmp 170KB 91.bmp 170KB
model-140002.data-00000-of-00001 7.65MB BEPS.iml 453B model-140002.index 4KB
20.jpg 360KB 1.jpg 285KB mosaicfloor.jpg 222KB 20.jpg 191KB 24.jpg 186KB 89.jpg 180KB 89.jpg 164KB 90.jpg 160KB 88.jpg 154KB 88.jpg 148KB crossstitch.jpg 143KB 13_19.jpg 142KB 02_01.jpg 140KB graffiti.jpg 138KB 90.jpg 137KB 02_02.jpg 133KB 02_01.jpg 133KB risk.jpg 123KB 37.jpg 123KB 11_12.jpg 120KB 01_15.jpg 108KB 89.jpg 102KB Bishapur_zan.jpg 95KB 24.jpg 92KB 91.jpg 92KB 37.jpg 89KB 01_04.jpg 85KB 35010.jpg 81KB 90.jpg 76KB 88.jpg 72KB 23084.jpg 70KB 89.jpg 68KB 02_01.jpg 68KB 89.jpg 68KB 37073.jpg 65KB 02_02.jpg 61KB 90.jpg 58KB 02_02.jpg 57KB 02_01.jpg 56KB 91.jpg 54KB 88.jpg 54KB 02_01.jpg 52KB 01_15.jpg 51KB 01_15.jpg 49KB 89.jpg 48KB 88.jpg 45KB 01_04.jpg 44KB 90.jpg 38KB 01_04.jpg 29KB 90.jpg 28KB 88.jpg 25KB 02_01.jpg 17KB README.md 1KB model-140002.meta 1.04MB
img_5089.png 408KB img_5014.png 371KB img_5076.png 315KB 12.png 260KB test.prototxt 9KB vgg16.prototxt 8KB vgg16.pth 78.23MB bdcn_120000.pth 62.22MB bdcn_pretrained_on_bsds500.pth 62.21MB bdcn_pretrained_on_nyudv2_rgb.pth 62.21MB model45_2.pth 2.16MB model_vdsr_ReguTerm.py 25KB utils2.py 19KB ablation.py 14KB ablation_train.py 11KB train.py 11KB bdcn.py 9KB BEPS_train_by_TED.py 7KB ablation_vgg16_c.py 5KB vgg16_c.py 5KB module.py 4KB ablation_test.py 4KB cfg.py 3KB dataset.py 3KB test.py 3KB test_ms.py 3KB test_DTFED.py 3KB共 125 条
- 1
- 2
资源评论
