implement segformer model

Doodleverse · Feb 11, 2023 · 1e26ef8 · 1e26ef8
1 parent b66ff15
commit 1e26ef8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doodleverse_utils/prediction_imports.py b/doodleverse_utils/prediction_imports.py
@@ -126,7 +126,7 @@ def seg_file2tensor_3band(f, TARGET_SIZE):
 
 # =========================================================
 def do_seg(
-    f, M, metadatadict, sample_direc, 
+    f, M, metadatadict, MODEL, sample_direc, 
     NCLASSES, N_DATA_BANDS, TARGET_SIZE, TESTTIMEAUG, WRITE_MODELMETADATA,
     OTSU_THRESHOLD,
     out_dir_name='out'