dvc.lock

schema: '2.0'
stages:
  data_split:
    cmd: python src/data_split.py
    deps:
    - path: data/all_data/
      hash: md5
      md5: 15f848fc495bd9d650462e65288f286c.dir
      size: 157899315
      nfiles: 317
    - path: src/data_split.py
      hash: md5
      md5: 8d802fe12f83502b7c7c0c012e469f4f
      size: 3465
    params:
      params.yaml:
        base:
          random_seed: 0
          model_image_size: 256
          loss_function: iou_loss
        data_split:
          test_split: 0.2
          data_dir: data/all_data/
          train_data_dir: data/train/
          test_data_dir: data/test/
    outs:
    - path: data/test
      hash: md5
      md5: e8766d24ccc24dcc2c3d7c2411524b01.dir
      size: 28969711
      nfiles: 64
    - path: data/train
      hash: md5
      md5: b863265b49281c883ce49f3d3365369a.dir
      size: 128923456
      nfiles: 252
  train:
    cmd: python src/train.py
    deps:
    - path: data/train/
      hash: md5
      md5: b863265b49281c883ce49f3d3365369a.dir
      size: 128923456
      nfiles: 252
    - path: src/train.py
      hash: md5
      md5: 1ebaa8f71da82bc67cdd52dab1680b73
      size: 8376
    - path: src/unet.py
      hash: md5
      md5: 1f9fb5155f1dfde0bc61a7aabc8e3e3f
      size: 6864
    params:
      params.yaml:
        base:
          random_seed: 0
          model_image_size: 256
          loss_function: iou_loss
        train:
          train_data_dir: data/train/
          model_save_dir: models/
          batch_size: 5
          epochs: 5
          activation_function: sigmoid
          learning_rate: 0.0001
          norm_upper_bound: 7
          norm_lower_bound: -1
          validation_split: 0.2
    outs:
    - path: models/catsnet_model.keras
      hash: md5
      md5: 5eb2e3bbbe773dc2aaaa1b4e7ce678c3
      size: 23455501
    - path: results/train/
      hash: md5
      md5: 27b2a0a061ed8f2b0f35714ad91e25f7.dir
      size: 678
      nfiles: 5
  evaluate:
    cmd: python src/evaluate.py
    deps:
    - path: data/test/
      hash: md5
      md5: e8766d24ccc24dcc2c3d7c2411524b01.dir
      size: 28969711
      nfiles: 64
    - path: models/catsnet_model.keras
      hash: md5
      md5: 5eb2e3bbbe773dc2aaaa1b4e7ce678c3
      size: 23455501
    - path: src/evaluate.py
      hash: md5
      md5: 2569536eb4e03562b5f48875b6b881df
      size: 6215
    - path: src/unet.py
      hash: md5
      md5: 1f9fb5155f1dfde0bc61a7aabc8e3e3f
      size: 6864
    params:
      params.yaml:
        base:
          random_seed: 0
          model_image_size: 256
          loss_function: iou_loss
        evaluate:
          model_path: models/catsnet_model.keras
          test_data_dir: data/test/
    outs:
    - path: results/evaluate/
      hash: md5
      md5: 808749d6794fb63a556b938ca2572a73.dir
      size: 3182997
      nfiles: 17