13hannes11
/

master_thesis_models

focus-prediction

Model card Files Files and versions Metrics Training metrics Community

Hannes Kuchelmeister commited on Apr 29, 2022

Commit

1a31ea8

·

1 Parent(s): d5c44c3

fix mdct (add square)

Files changed (1) hide show

src/models/focus_traditional.py +1 -1

src/models/focus_traditional.py CHANGED Viewed

@@ -37,7 +37,7 @@ def midfrequency_dct(img):
     )
     img_grey = torch.mean(img, dim=0).unsqueeze(0)
-    filtered = kornia.filters.filter2d(img_grey, kernel)
     sum = torch.sum(filtered)
     return 100 / sum

     )
     img_grey = torch.mean(img, dim=0).unsqueeze(0)
+    filtered = torch.square(kornia.filters.filter2d(img_grey, kernel))
     sum = torch.sum(filtered)
     return 100 / sum