forked from DedInc/gamma-ai-watermark-remover
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
10 lines (9 loc) · 4.74 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
class WatermarkConfig:
SIMILARITY_THRESHOLD = 0.95
SIMILARITY_INVERTED_THRESHOLD = 0.95
TEMPLATE_HISTOGRAM = [0.1886512218343383, 0.010942557918121232, 0.00925420501428118, 0.010663281497937162, 0.00974928594097112, 0.006639162170739447, 0.0048873373532212, 0.004100285623611552, 0.0033132338940019043, 0.002564265312599175, 0.0023103776578863853, 0.0025769596953348142, 0.0021326562995874327, 0.001612186607426214, 0.001827991113932085, 0.001294827039035227, 0.0012059663598857505, 0.0013582989527134244, 0.0017518248175182481, 0.0055220564900031735, 0.0027419866708981275, 0.0016502697556331322, 0.0011678832116788322, 0.0013582989527134244, 0.0011171056807362742, 0.0009774674706442398, 0.0011805775944144716, 0.0013329101872421453, 0.0011678832116788322, 0.0009774674706442398, 0.001002856236115519, 0.0010790225325293557, 0.0010536337670580768, 0.0011551888289431927, 0.0010536337670580768, 0.0011171056807362742, 0.0011171056807362742, 0.001536020311012377, 0.01976515391939067, 0.021085369723897174, 0.035061885115836244, 0.03653443351317042, 0.03754998413202158, 0.03904792129482704, 0.019854014598540148, 0.01783560774357347, 0.01781021897810219, 0.016743890828308475, 0.01679466835925103, 0.015550618851158363, 0.022494446207553156, 0.032078705172960964, 0.03847667407172326, 0.0405458584576325, 0.037867343700412566, 0.0405458584576325, 0.04178990796572517, 0.022494446207553156, 0.010688670263408442, 0.00463344969850841, 0.0039606474135195175, 0.0023357664233576644, 0.0017010472865756903, 0.0012694382735639479, 0.0009139955569660425, 0.0007616629641383687, 0.000926689939701682, 0.0007489685814027293, 0.0009139955569660425, 0.0006474135195176135, 0.000660107902253253, 0.0005331640748968582, 0.00045699777848302127, 0.0007235798159314504, 0.0005585528403681371, 0.0004316090130117423, 0.0006220247540463345, 0.0006347191367819739, 0.0007616629641383687, 0.0005839416058394161, 0.0005712472231037765, 0.0005966359885750556, 0.00048238654395430024, 0.000609330371310695, 0.0004316090130117423, 0.0006474135195176135, 0.0007489685814027293, 0.000660107902253253, 0.000660107902253253, 0.0005712472231037765, 0.0003681370993335449, 0.0004443033957473818, 0.0005204696921612187, 0.000660107902253253, 0.0005966359885750556, 0.0006347191367819739, 0.0006347191367819739, 0.0006981910504601713, 0.0006981910504601713, 0.0004316090130117423] + ([0] * (256 - 100))
INVERTED_TEMPLATE_HISTOGRAM = [0.19600126943827356, 0.010345921929546175, 0.009927007299270074, 0.009431926372580134, 0.0077435734687400825, 0.005242780069819105, 0.003973341796255157, 0.003211678832116788, 0.0028435417327832436, 0.0021072675341161536, 0.0023103776578863853, 0.002488099016185338, 0.002069184385909235, 0.001561409076483656, 0.0015994922246905744, 0.0011678832116788322, 0.0011171056807362742, 0.0011171056807362742, 0.0009393843224373214, 0.0008886067914947636, 0.0009013011742304031, 0.0007997461123452872, 0.000926689939701682, 0.001002856236115519, 0.0008251348778165661, 0.0008251348778165661, 0.0006220247540463345, 0.001002856236115519, 0.0008251348778165661, 0.000660107902253253, 0.0006981910504601713, 0.0007616629641383687, 0.0006981910504601713, 0.0005966359885750556, 0.000660107902253253, 0.000660107902253253, 0.0006347191367819739, 0.0005966359885750556, 0.0007108854331958109, 0.0006347191367819739, 0.0007616629641383687, 0.0004950809266899397, 0.0005585528403681371, 0.0005077753094255791, 0.0007235798159314504, 0.0004950809266899397, 0.0005458584576324976, 0.0005204696921612187, 0.0005712472231037765, 0.0007235798159314504, 0.0005966359885750556, 0.0007870517296096477, 0.0011805775944144716, 0.0017137416693113296, 0.0026150428435417327, 0.003986036178990796, 0.006651856553475087, 0.009381148841637576, 0.006448746429704856, 0.005471278959060616, 0.0035036496350364962, 0.0026785147572199303, 0.0017137416693113296, 0.00145985401459854, 0.0013456045699777848, 0.00145985401459854, 0.0013582989527134244, 0.0010917169152649953, 0.0010409393843224374, 0.0010917169152649953, 0.0007997461123452872, 0.001320215804506506, 0.002830847350047604, 0.0033259282767375436, 0.0019676293240241194, 0.0007362741986670898, 0.0006854966677245319, 0.0009013011742304031, 0.0011678832116788322, 0.0009901618533798794, 0.0010790225325293557, 0.0009647730879086005, 0.0010282450015867979, 0.0007870517296096477, 0.0008251348778165661, 0.0008886067914947636, 0.0009774674706442398, 0.001142494446207553, 0.0009774674706442398, 0.000952078705172961, 0.000926689939701682, 0.0007362741986670898, 0.0008251348778165661, 0.0005712472231037765, 0.0006474135195176135, 0.0006981910504601713, 0.0006474135195176135, 0.0007108854331958109, 0.0005712472231037765, 0.00048238654395430024] + ([0] * (256 - 100))
@staticmethod
def get_template_histogram_array():
import numpy as np
return np.array(WatermarkConfig.TEMPLATE_HISTOGRAM)