Image-classification-Deep-Learning-Project/dvc.lock at main · PrathikVijaykumar/Image-classification-Deep-Learning-Project · 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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/cnnClassifier/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: d916c32278bbb44d8bd6bd082f6cc0a2
      size: 558
    - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: 28998e02a534050f6d2f211853b523cf
      size: 890
    outs:
    - path: artifacts/data_ingestion/kidney_scan_images
      hash: md5
      md5: 96e259aaa7f20f6aace57fa0b3473b29.dir
      size: 63535339
      nfiles: 500
  prepare_base_model:
    cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: d916c32278bbb44d8bd6bd082f6cc0a2
      size: 558
    - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: fa608a843d4ca311728f3dcedf554815
      size: 989
    params:
      params.yaml:
        CLASSES: 2
        IMAGE_SIZE:
        - 224
        - 224
        - 3
        INCLUDE_TOP: false
        LEARNING_RATE: 0.04
        WEIGHTS: imagenet
    outs:
    - path: artifacts/prepare_base_model
      hash: md5
      md5: b0c07f9ffeffff53ea5f26cfa8805a00.dir
      size: 118054560
      nfiles: 2
  training:
    cmd: python src/cnnClassifier/pipeline/stage_03_model_training.py
    deps:
    - path: artifacts/data_ingestion/kidney_scan_images
      hash: md5
      md5: 96e259aaa7f20f6aace57fa0b3473b29.dir
      size: 63535339
      nfiles: 500
    - path: artifacts/prepare_base_model
      hash: md5
      md5: b0c07f9ffeffff53ea5f26cfa8805a00.dir
      size: 118054560
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: d916c32278bbb44d8bd6bd082f6cc0a2
      size: 558
    - path: src/cnnClassifier/pipeline/stage_03_model_training.py
      hash: md5
      md5: f92e35047b7f904ef1c5cbd482a96fad
      size: 917
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 5
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: b98843f7053657313379951053577b6c
      size: 59337520
  evaluation:
    cmd: python src/cnnClassifier/pipeline/stage_04_model_evaluation.py
    deps:
    - path: artifacts/data_ingestion/kidney_scan_images
      hash: md5
      md5: 96e259aaa7f20f6aace57fa0b3473b29.dir
      size: 63535339
      nfiles: 500
    - path: artifacts/training/model.h5
      hash: md5
      md5: b98843f7053657313379951053577b6c
      size: 59337520
    - path: config/config.yaml
      hash: md5
      md5: d916c32278bbb44d8bd6bd082f6cc0a2
      size: 558
    - path: src/cnnClassifier/pipeline/stage_04_model_evaluation.py
      hash: md5
      md5: 788e3b4f442554a38f12891217af9ce4
      size: 1022
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 90ed20647dbc426436ea4272d6ab0084
      size: 58