File size: 454 Bytes
e2855de
 
 
 
2add545
 
e2855de
 
9a6ea32
 
e2855de
 
 
 
 
 
 
 
9a6ea32
e2855de
5606920
9a6ea32
5606920
 
e2855de
 
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
model:
  backbone: resnet50
  num_classes: 2
  pretrained: true
  learning_rate:  0.0050 
  epochs: 10
  batch_size: 8
  optimizer: adam
  image_sample_size: 10000
  trainable_backbone_layers: 1

dataset:
  image_dir: datasets/images
  annotation_dir: datasets/annotations
  train_split: 0.8
  val_split: 0.2

notebooks:
  visualization: logs/training_log.json

logs:
  log_dir: logs/training_log.txt
  log_interval: 10

evaluation:
  iou_threshold: 0.5