dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python src\cnnImageClassifier\pipeline\stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 7ec9aff8fea350a2d8d46c949d7961d6
      size: 584
    - path: src\cnnImageClassifier\pipeline\stage_01_data_ingestion.py
      hash: md5
      md5: 5d798f0920e9e5a12799193c1db48f98
      size: 911
    outs:
    - path: artifacts/data_ingestion/PTE_HDPE_BOTTLS_DATASET
      hash: md5
      md5: 146d1cb7bf6b93ecac04aa5cfb045161.dir
      size: 132606488
      nfiles: 3762
  prepare_base_model:
    cmd: python src/cnnImageClassifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 7ec9aff8fea350a2d8d46c949d7961d6
      size: 584
    - path: src/cnnImageClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: d3c37456655f015ffa55f56da2605b56
      size: 1014
    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: 9ee0e17aeb5eabd7815fd1c9bd1a116d.dir
      size: 118054560
      nfiles: 2
  training:
    cmd: python src/cnnImageClassifier/pipeline/stage_03_model_training.py
    deps:
    - path: artifacts/data_ingestion/PTE_HDPE_BOTTLS_DATASET
      hash: md5
      md5: 146d1cb7bf6b93ecac04aa5cfb045161.dir
      size: 132606488
      nfiles: 3762
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 9ee0e17aeb5eabd7815fd1c9bd1a116d.dir
      size: 118054560
      nfiles: 2
    - path: config/config.yaml
      hash: md5
      md5: 7ec9aff8fea350a2d8d46c949d7961d6
      size: 584
    - path: src/cnnImageClassifier/pipeline/stage_03_model_training.py
      hash: md5
      md5: 0c0cd5a1eb308bc61fdb5e004f712cb4
      size: 942
    params:
      params.yaml:
        AUGMENTATION: true
        BATCH_SIZE: 16
        EPOCHS: 3
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: artifacts/training/model.h5
      hash: md5
      md5: f789949d43fff2fc02ac2f2d619b1e65
      size: 59337520
  evaluation:
    cmd: python src\cnnImageClassifier\pipeline\stage_04_model_eveluation_mlflow.py
    deps:
    - path: artifacts/data_ingestion/PTE_HDPE_BOTTLS_DATASET
      hash: md5
      md5: 146d1cb7bf6b93ecac04aa5cfb045161.dir
      size: 132606488
      nfiles: 3762
    - path: artifacts/training/model.h5
      hash: md5
      md5: f789949d43fff2fc02ac2f2d619b1e65
      size: 59337520
    - path: config/config.yaml
      hash: md5
      md5: 7ec9aff8fea350a2d8d46c949d7961d6
      size: 584
    - path: src\cnnImageClassifier\pipeline\stage_04_model_eveluation_mlflow.py
      hash: md5
      md5: 1e9ddfc2d53af1a82c99967100d13b6f
      size: 1008
    params:
      params.yaml:
        BATCH_SIZE: 16
        IMAGE_SIZE:
        - 224
        - 224
        - 3
    outs:
    - path: scores.json
      hash: md5
      md5: 771e64b4aa8b05aab96f830202b83803
      size: 72