SuperResolution/dvc.yaml at main · Karan-Choudhary/SuperResolution · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
stages:
  get_data:
    cmd: python src\get_data.py --config=params.yaml
    deps:
      - src\read_params.py
      - src\get_data.py
      - src\utils\preprocessing.py
      - data\train\HR
      - data\validation\HR
      - data\validation\LR

  train_model:
    cmd: python src\train_model.py --config=params.yaml
    deps:
      - src\get_data.py
      - src\train_model.py
      - src\model\losses\loss.py
      - src\model\Subpixel\Subpixel_conv2D.py
      - src\model\build.py
      - src\model\models.py
    params:
    - data_load.batch_size
    - data_load.buffer_size
    - data_load.HR_img_width
    - data_load.HR_img_height
    - data_load.LR_img_width
    - data_load.LR_img_height
    - model.num_res_blocks
    - model.num_filters
    - model.kernel_size
    - model.num_channels
    - model.upsampling_factor
    - model.strides
    - model.padding
    - training.epochs
    - loss.lambda
    metrics:
    - report\params.json:
        cache: false
    - report\logs.json:
        cache: false
    outs:
    - saved_models\Generator.h5
    - saved_models\Discriminator.h5

  evaluate_model:
    cmd: python src\evaluate_model.py --config=params.yaml
    deps:
    - src\display\showOutput.py
    - src\evaluate.py
    - saved_models\Generator.h5
    params:
    - test.num_samples
    outs:
    - Results