Tuberculosis-Vision-Scan/dvc.lock at main · VivekShinde7/Tuberculosis-Vision-Scan · 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: 1e19083c6c6851b9662e9b6fe403322a
      size: 580
    - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: 23963f14a622cf0e5db6ec3941a259fc
      size: 907
    outs:
    - path: artifacts/data_ingestion/TB_Chest_Radiography_Database
      hash: md5
      md5: edc0df46ce62ded330caa56896046380.dir
      size: 694671204
      nfiles: 4203
  prepare_base_model:
    cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 1e19083c6c6851b9662e9b6fe403322a
      size: 580
    - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: fe276034f349392a04d69bc9633ceb3c
      size: 995
    params:
      params.yaml:
        CLASSES: 2
        IMAGE_SIZE:
        - 224
        - 224
        - 3
        INCLUDE_TOP: false
        LEARNING_RATE: 0.01
        WEIGHTS: imagenet
    outs:
    - path: artifacts/prepare_base_model
      hash: md5
      md5: e14874d87648d4b0133ef5415fa9da6f.dir
      size: 118054560
      nfiles: 2
  training:
    cmd: python src/cnnClassifier/pipeline/stage_03_model_trainer.py
    deps:
    - path: artifacts/data_ingestion/TB_Chest_Radiography_Database
      hash: md5
      md5: edc0df46ce62ded330caa56896046380.dir
      size: 694671204
      nfiles: 4203
    - path: artifacts/prepare_base_model
      hash: md5
      md5: e14874d87648d4b0133ef5415fa9da6f.dir
      size: 118054560
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: 1e19083c6c6851b9662e9b6fe403322a
      size: 580
    - path: src/cnnClassifier/pipeline/stage_03_model_trainer.py
      hash: md5
      md5: c4201629623cac473b4e6e41ed61270b
      size: 933
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 10
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: 98284fdffffa9527d8cb50d42de6ac06
      size: 59337520
  evaluation:
    cmd: python src/cnnClassifier/pipeline/stage_04_model_evaluation.py
    deps:
    - path: artifacts/data_ingestion/TB_Chest_Radiography_Database
      hash: md5
      md5: edc0df46ce62ded330caa56896046380.dir
      size: 694671204
      nfiles: 4203
    - path: artifacts/training/model.h5
      hash: md5
      md5: 98284fdffffa9527d8cb50d42de6ac06
      size: 59337520
    - path: config/config.yaml
      hash: md5
      md5: 1e19083c6c6851b9662e9b6fe403322a
      size: 580
    - path: src/cnnClassifier/pipeline/stage_04_model_evaluation.py
      hash: md5
      md5: 3fceb4ac4bd09ba9900e1833534af160
      size: 936
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 660229b964240d4b40c13baa11a5f297
      size: 72