cmarkea
/

detr-layout-detection

Object Detection

image-segmentation

Model card Files Files and versions

Cyrile commited on Aug 14, 2024

Commit

eefdbff

·

verified ·

1 Parent(s): 2e95895

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -68,7 +68,7 @@ with torch.inference_mode():
 threshold=0.4
 segmentation_mask = img_proc.post_process_segmentation(
-    out_seg,
     threshold=threshold,
     target_sizes=[img.size[::-1]]
 )

 threshold=0.4
 segmentation_mask = img_proc.post_process_segmentation(
+    output,
     threshold=threshold,
     target_sizes=[img.size[::-1]]
 )