Create deephashloss.py

2021-08-25 13:59:54 +08:00 · 2021-08-25 13:59:54 +08:00 · 9323b1478a
parent 52663edf60
commit 9323b1478a
1 changed files with 6 additions and 0 deletions
--- a/ppcls/loss/deephashloss.py
+++ b/ppcls/loss/deephashloss.py
@ -0,0 +1,6 @@
+            # do binarize
+            if self.config["Global"].get("feature_binarize") == "round":
+                batch_feas = paddle.round(batch_feas).astype("float32") * 2.0 - 1.0
+
+            if self.config["Global"].get("feature_binarize") == "sign":
+                batch_feas = paddle.sign(batch_feas).astype("float32")