Upload 6 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+model.engine filter=lfs diff=lfs merge=lfs -text

config.yaml

@@ -0,0 +1,111 @@
+project_name: Nordic_Multicentury
+exp_name: Nordic_Multicentury
+exp: ${exp_name}_${now_dir}
+model_name: s
+task: segment
+train:
+  root: /home/incognito/AI/D-FINE-seg
+  pretrained_dataset: coco
+  pretrained_model_path: pretrained/dfine_seg_s_coco.pt
+  data_path: ${train.root}/datasets/Nordic_Multicentury
+  path_to_test_data: ${train.root}/images_nordic
+  path_to_save: ${train.root}/output/models/${exp}
+  debug_img_path: ${train.root}/output/debug_images
+  eval_preds_path: ${train.root}/output/eval_preds
+  bench_img_path: ${train.root}/output/bench_imgs
+  infer_path: ${train.root}/output/infer
+  use_wandb: false
+  device: cuda
+  label_to_name:
+    0: textline
+  use_one_class: false
+  ddp:
+    enabled: false
+    n_gpus: 2
+  decision_metrics:
+  - f1
+  img_size:
+  - 768
+  - 1152
+  keep_ratio: false
+  to_visualize_eval: true
+  debug_img_processing: true
+  amp_enabled: true
+  clip_max_norm: 0.1
+  batch_size: 4
+  b_accum_steps: 2
+  epochs: 500
+  early_stopping: 10
+  ignore_background_epochs: 0
+  num_workers: 2
+  mask_batch_size: 1000
+  conf_thresh: 0.5
+  iou_thresh: 0.5
+  use_ema: true
+  ema_momentum: 0.9998
+  use_scheduler: true
+  base_lr: ${train.lrs.${model_name}.base_lr}
+  backbone_lr: ${train.lrs.${model_name}.backbone_lr}
+  cycler_pct_start: 0.1
+  weight_decay: 0.000125
+  betas:
+  - 0.9
+  - 0.999
+  label_smoothing: 0.0
+  mosaic_augs:
+    mosaic_prob: 0.0
+    no_mosaic_epochs: 5
+    mosaic_scale:
+    - 0.5
+    - 1.5
+    degrees: 0.0
+    translate: 0.2
+    shear: 2.0
+  augs:
+    rotation_degree: 10
+    rotation_p: 0.0
+    multiscale_prob: 0.0
+    rotate_90: 0.05
+    left_right_flip: 0.3
+    up_down_flip: 0.0
+    to_gray: 0.01
+    blur: 0.01
+    gamma: 0.02
+    brightness: 0.02
+    noise: 0.01
+    coarse_dropout: 0.0
+  seed: 42
+  cudnn_fixed: false
+  lrs:
+    'n':
+      backbone_lr: 0.0004
+      base_lr: 0.0008
+    s:
+      backbone_lr: 6.0e-05
+      base_lr: 0.00025
+    m:
+      backbone_lr: 2.0e-05
+      base_lr: 0.00015
+    l:
+      backbone_lr: 6.25e-06
+      base_lr: 0.000125
+    x:
+      backbone_lr: 1.5e-06
+      base_lr: 0.0001
+split:
+  ignore_negatives: false
+  shuffle: true
+  train_split: 0.84
+  val_split: 0.15
+  test_split: 0.01
+export:
+  half: true
+  max_batch_size: 1
+  dynamic_input: false
+  ov_int8_max_drop: 0.01
+infer:
+  to_crop: true
+  paddings:
+    w: 0.05
+    h: 0.05
+now_dir: ${now:%Y-%m-%d}

model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0b657f4d47a5fd7cce39802a807ccb04aaf506d8f0638e4f69a22f5f93e4bd6
+size 44952860

model.engine

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:314a38fcb69ed7b57eeb87b38616b1395e1c3782ba3a61f8e870ed69ac95e166
+size 31429748

model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86c52c8f5cfc8bf33aee645ff050ac5dc46769b750b11186ef21f34b6be23d9c
+size 45067285

model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:58299d29e52c588f88207908aecf4fc5422f361ac7a03ce3b061c4a8c495de71
+size 44488443