Tweak the ML partition aggressiveness
STATS_CHANGED
Change-Id: Ic857b36c94d405b99cd2b063af3dc52231afead7
diff --git a/av1/encoder/partition_model_weights.h b/av1/encoder/partition_model_weights.h
index 493d893..576b981 100644
--- a/av1/encoder/partition_model_weights.h
+++ b/av1/encoder/partition_model_weights.h
@@ -2622,61 +2622,63 @@
#define FEATURE_SIZE 4
// Mean and std
static const float av1_hd_partition_breakout_nn_mean_128[FEATURE_SIZE] = {
- 10.730792598027795f,
- 376.6180794798389f,
- 699.9787168227473f,
- 2069.426780349039f,
+ 1.7569518370024242f,
+ 5.5324198861333107f,
+ 5.5260479508123526f,
+ 7.1913091289363038f,
};
static const float av1_hd_partition_breakout_nn_std_128[FEATURE_SIZE] = {
- 29.023640354060735f, 659.3967891826734f, 940.2882285903872f,
- 2566.2070264789245f
+ 1.3286579037156365f, 1.4302357105525316f, 1.947275803679912f,
+ 1.4856880036033451f
};
static const float av1_hd_partition_breakout_nn_mean_64[FEATURE_SIZE] = {
- 22.220171631578605f,
- 360.49969866754554f,
- 642.6605866566223f,
- 1022.2158389850164f,
+ 2.5083185600034166f,
+ 5.5459875545856132f,
+ 5.4419578738009537f,
+ 6.3575563591302293f,
};
static const float av1_hd_partition_breakout_nn_std_64[FEATURE_SIZE] = {
- 69.0770174031301f, 758.5544028348198f, 891.8245485291498f, 1901.3873859786233f
+ 1.5327874427371651f, 1.4413133143150851f, 1.852577618179309f,
+ 1.7441221691208793f
};
static const float av1_hd_partition_breakout_nn_mean_32[FEATURE_SIZE] = {
- 56.23900643356807f,
- 411.708483646884f,
- 552.7719521312714f,
- 671.8117635469166f,
+ 3.2925380941946631f,
+ 5.6592503897046313f,
+ 5.2656875189877344f,
+ 5.88623228906336f,
};
static const float av1_hd_partition_breakout_nn_std_32[FEATURE_SIZE] = {
- 198.97960359590587f, 959.3829390828911f, 830.9980036115545f,
- 1498.4982976375336f
+ 1.6894083970930263f, 1.4685304613224999f, 1.8598497961630693f,
+ 1.7648307635114531f
};
static const float av1_hd_partition_breakout_nn_mean_16[FEATURE_SIZE] = {
- 136.9843430416749f,
- 532.9528184025893f,
- 504.7980827006724f,
- 445.7892729738405f,
+ 4.334899946924148f,
+ 5.8802354638427667f,
+ 5.2098555971571239f,
+ 5.5808807097219715f,
};
static const float av1_hd_partition_breakout_nn_std_16[FEATURE_SIZE] = {
- 478.52767648003527f, 1308.8412547684493f, 795.1318845991449f,
- 1176.5665198192673f
+ 1.7269713123782717f, 1.5227226730026808f, 1.9255508380289716f,
+ 1.6959046024771494f
};
static const float av1_hd_partition_breakout_nn_mean_8[FEATURE_SIZE] = {
- 275.0553416698255f,
- 566.7785552508784f,
- 406.48781378761095f,
- 247.1067473273997f,
+ 5.17688933331104f,
+ 6.0068390263641165f,
+ 4.9884361306237377f,
+ 5.0973575838899237f,
};
static const float av1_hd_partition_breakout_nn_std_8[FEATURE_SIZE] = {
- 935.5925922140688f, 1435.299322400821f, 678.9806893808023f, 791.3637162723987f
+ 1.693000978888298f, 1.4741850818874491f, 1.8398258859567682f,
+ 1.6434668808089992f
};
static const float *const av1_hd_partition_breakout_nn_mean[5] = {
@@ -2693,418 +2695,397 @@
static const float
av1_hd_partition_breakout_nn_weights_128_layer0[FEATURE_SIZE * 32] = {
- 0.4625445008277893f, 0.046699944883584976f,
- 0.01686757430434227f, -0.20717839896678925f,
- 3.7204904556274414f, -0.09939002245664597f,
- -0.03257349878549576f, -0.02249263785779476f,
- 4.27296257019043f, -0.2201482355594635f,
- -0.02630791626870632f, -0.8290131092071533f,
- 4.106664657592773f, -0.03872422128915787f,
- -0.0008374569588340819f, 0.008610394783318043f,
- 2.8612468242645264f, 0.029475420713424683f,
- 0.01878979615867138f, -0.07678258419036865f,
- 2.642153024673462f, -0.7023047208786011f,
- -0.16276267170906067f, -1.7577937841415405f,
- 3.9672751426696777f, -0.019311338663101196f,
- -0.00475256098434329f, 7.173077028710395e-05f,
- -0.06531551480293274f, -0.07583651691675186f,
- -0.02518392540514469f, 0.005303085781633854f,
- 4.6276326179504395f, -0.4294697642326355f,
- -0.06581428647041321f, -0.4343421459197998f,
- -3.5704429149627686f, -0.3216317892074585f,
- 0.2882029414176941f, 0.6410080194473267f,
- -3.4585084915161133f, 0.24044956266880035f,
- 0.2667864263057709f, 0.35314950346946716f,
- -0.9072761535644531f, 0.4799148142337799f,
- -1.9291492700576782f, -0.14138458669185638f,
- -4.956376075744629f, -0.4101237654685974f,
- -0.00801194179803133f, 0.016341974958777428f,
- 10.001052856445312f, -0.10566182434558868f,
- -0.008026782423257828f, -0.04508136212825775f,
- -0.11174708604812622f, -0.1031208261847496f,
- 0.004450094420462847f, -0.0714421346783638f,
- -3.7611849308013916f, -0.42005446553230286f,
- -0.39350515604019165f, 0.47153064608573914f,
- -0.36057817935943604f, -2.4768900871276855f,
- -0.0051008667796850204f, 0.588866114616394f,
- 3.6189825534820557f, -0.018487676978111267f,
- 0.033942464739084244f, 0.3527982831001282f,
- 2.350560426712036f, -1.840644359588623f,
- 0.1856888085603714f, -1.4500601291656494f,
- -2.9771881103515625f, 0.15055160224437714f,
- -0.7389538884162903f, -0.02373156137764454f,
- 3.244478464126587f, 0.09777998179197311f,
- 0.03143433853983879f, -0.3365783095359802f,
- -3.579131841659546f, -0.3401176333427429f,
- 0.3683990240097046f, 0.6233412027359009f,
- -5.737433433532715f, -1.0643895864486694f,
- -0.0862712413072586f, -0.8468044400215149f,
- 4.366209983825684f, -0.03896212950348854f,
- -0.005277127958834171f, -0.054806072264909744f,
- 4.20292854309082f, -0.33671531081199646f,
- -0.04025593772530556f, -0.4954434335231781f,
- 1.4843676090240479f, -0.05833450332283974f,
- -1.2679489850997925f, 0.526577353477478f,
- 4.186404705047607f, -0.019188573583960533f,
- 0.03789565712213516f, 0.35230082273483276f,
- 3.3375051021575928f, -0.01609140820801258f,
- 0.027185490354895592f, 0.25255802273750305f,
- 4.186381816864014f, -0.10051476955413818f,
- -0.03126750886440277f, -0.02391117438673973f,
- 3.763073205947876f, -0.04791776463389397f,
- -2.3893446268630214e-05f, 0.010260583832859993f,
- -2.701782464981079f, -0.3117542862892151f,
- -0.08752211928367615f, 0.4682832360267639f,
- 2.9983580112457275f, -0.47988685965538025f,
- 0.2418350726366043f, 0.451855331659317f
+ -0.24421754479408264f, 0.5826895833015442f, -0.468494713306427f,
+ 0.05104721337556839f, -0.3030756115913391f, -0.47995227575302124f,
+ 0.30220189690589905f, 0.6468079686164856f, -0.6581971049308777f,
+ -0.08561690151691437f, -0.25164133310317993f, -0.7251309752464294f,
+ 0.28622835874557495f, -0.3805813491344452f, -1.2038640975952148f,
+ -0.03289572894573212f, -0.37501949071884155f, -0.6697373390197754f,
+ 0.10718680173158646f, 0.39980190992355347f, -0.6406639218330383f,
+ 0.3059859871864319f, -0.04687154293060303f, -0.0283386018127203f,
+ 0.2230420708656311f, -0.024931879714131355f, 0.029338961467146873f,
+ -0.39070868492126465f, 0.12055488675832748f, 0.44709470868110657f,
+ -0.2240329086780548f, 0.3899381160736084f, 0.7343757748603821f,
+ 0.3804762363433838f, -0.6964248418807983f, 0.031393926590681076f,
+ 1.1329405307769775f, 0.2145451456308365f, 0.15080654621124268f,
+ -0.5583563446998596f, 0.7352061867713928f, 0.03883780539035797f,
+ -0.003559485776349902f, -0.6891534328460693f, -0.2841552793979645f,
+ -0.033437542617321014f, -0.05212170630693436f, 0.46065035462379456f,
+ 0.49060872197151184f, -1.0210745334625244f, -0.20526815950870514f,
+ 0.3765658140182495f, -0.050877176225185394f, -0.4010245203971863f,
+ 0.2002274990081787f, -0.45646876096725464f, 0.028925752267241478f,
+ 0.44919347763061523f, -0.02976205199956894f, -0.14882135391235352f,
+ -0.23057465255260468f, -0.1957021802663803f, -0.02549680881202221f,
+ -0.8748094439506531f, -0.48586851358413696f, -0.4607398211956024f,
+ -0.19540899991989136f, 0.857029139995575f, -0.37086406350135803f,
+ -0.15118056535720825f, 0.18724219501018524f, 0.9177047610282898f,
+ -0.04173488914966583f, -0.3510979413986206f, 0.17673315107822418f,
+ -0.4069298505783081f, 0.020244762301445007f, -0.6126715540885925f,
+ 0.6862805485725403f, 0.1272497922182083f, -0.4310610890388489f,
+ -0.5232776403427124f, -0.26814699172973633f, 0.8925990462303162f,
+ -0.12301289290189743f, -0.5783265829086304f, 0.6845996379852295f,
+ 0.22650295495986938f, -0.11178003996610641f, -0.2689964771270752f,
+ -1.2071754932403564f, 0.8008237481117249f, -0.07050541788339615f,
+ -0.572569727897644f, 0.27917924523353577f, -0.6794210076332092f,
+ 0.37107107043266296f, 0.04119817912578583f, 0.08820059895515442f,
+ -0.29868319630622864f, -0.8464934229850769f, -0.6945635080337524f,
+ -0.6215496063232422f, 0.08221497386693954f, -0.6280669569969177f,
+ -0.44894012808799744f, 0.5003432631492615f, 0.4358958899974823f,
+ -0.03372211381793022f, 0.523205041885376f, -0.03965454921126366f,
+ -0.13546960055828094f, -0.3377343714237213f, -0.3924517333507538f,
+ -0.1991700977087021f, 0.6752629280090332f, -1.0135334730148315f,
+ 0.5485906600952148f, -0.020843572914600372f, -0.1461324244737625f,
+ -0.7540525794029236f, 0.21933406591415405f, -0.5565986633300781f,
+ 0.1918906718492508f, -0.40834516286849976f, 0.9896414279937744f,
+ 0.11563636362552643f, -0.017263229936361313f
};
static const float av1_hd_partition_breakout_nn_bias_128_layer0[32] = {
- -0.09511213004589081f, 0.9938786625862122f, 1.4839892387390137f,
- 1.0876423120498657f, 0.8019428253173828f, 0.2738432288169861f,
- 1.071292519569397f, -0.09138987958431244f, 1.2080857753753662f,
- -0.6339017748832703f, -0.4984428584575653f, -1.2238473892211914f,
- -1.3475229740142822f, 2.4443564414978027f, -0.16558663547039032f,
- 0.033367302268743515f, -0.7447078227996826f, 1.0401278734207153f,
- -0.4657363295555115f, -1.0536733865737915f, 0.8565444946289062f,
- -0.5986987948417664f, -0.789206862449646f, 1.1764090061187744f,
- 1.3952962160110474f, -0.17746582627296448f, 1.187064290046692f,
- 0.9370515942573547f, 1.1067490577697754f, 1.003738522529602f,
- -2.1517810821533203f, 0.8461123704910278f
+ 0.2676561176776886f, -0.29510897397994995f, -1.1071470975875854f,
+ 0.583503007888794f, -0.472633957862854f, -0.44106626510620117f,
+ 0.6721175909042358f, -0.42975232005119324f, -0.5145384073257446f,
+ 0.15512774884700775f, 1.0278737545013428f, -0.6500952839851379f,
+ 1.0237534046173096f, 0.5816664099693298f, 0.6633380055427551f,
+ -1.3322787284851074f, -0.4271796643733978f, 0.06402774155139923f,
+ 0.5194029808044434f, 0.03623528033494949f, -0.34109270572662354f,
+ -0.15831837058067322f, -0.07824128866195679f, 0.8782597184181213f,
+ -0.014673121273517609f, -0.20494380593299866f, -0.3248092532157898f,
+ 0.6707151532173157f, -0.2768913507461548f, -0.31312206387519836f,
+ -0.6461104154586792f, 0.16326004266738892f
};
static const float av1_hd_partition_breakout_nn_weights_128_layer1[32] = {
- -0.00859279278665781f, -2.9574432373046875f, -4.146053314208984f,
- -3.348489761352539f, -0.05143212527036667f, -2.0400538444519043f,
- -4.622243404388428f, -0.05732434615492821f, -5.16105318069458f,
- 1.1596481800079346f, 0.8911784291267395f, 8.114648818969727f,
- 19.255064010620117f, 11.491721153259277f, -0.2828827202320099f,
- 1.5688998699188232f, 0.7430379986763f, -1.0839532613754272f,
- -1.3526417016983032f, 5.244595527648926f, -4.02890682220459f,
- 0.8462737798690796f, 5.2530364990234375f, -4.688608646392822f,
- -4.285271644592285f, 0.3642231523990631f, -2.4636125564575195f,
- -0.8168565034866333f, -6.247287273406982f, -1.9640626907348633f,
- -2.2665669918060303f, -0.7841953039169312f
+ -0.31587469577789307f, 0.2444663941860199f, 1.1686184406280518f,
+ -1.136441707611084f, 0.462795227766037f, 0.16067592799663544f,
+ -0.08983717858791351f, -0.1526145339012146f, 1.2807552814483643f,
+ -1.8411308526992798f, -2.5061304569244385f, -0.33663204312324524f,
+ -1.3827414512634277f, -0.2779307961463928f, -0.47907790541648865f,
+ 1.9458348751068115f, 0.3252658247947693f, 0.6043151617050171f,
+ -0.2127833366394043f, 0.3176756501197815f, 0.9920678734779358f,
+ 0.49299460649490356f, 0.8988507390022278f, -0.7701797485351562f,
+ 0.29247382283210754f, 0.7251673340797424f, 0.460658997297287f,
+ -0.3683358430862427f, 0.6006630659103394f, 0.7680628299713135f,
+ 0.8999216556549072f, -0.529272735118866f
};
static const float av1_hd_partition_breakout_nn_bias_128_layer1[1] = {
- 0.1112925186753273f
+ -0.22755585610866547f
};
static const float
av1_hd_partition_breakout_nn_weights_64_layer0[FEATURE_SIZE * 32] = {
- -2.9600589275360107f, 0.042996086180210114f, -0.5422487258911133f,
- -0.16088658571243286f, 2.665553331375122f, 0.17293912172317505f,
- -0.011884346604347229f, -1.3571131229400635f, -4.151463508605957f,
- -0.09992470592260361f, -0.03328946977853775f, -0.19198642671108246f,
- -4.485296726226807f, 0.0004126991261728108f, 0.00528523325920105f,
- 0.0014443921390920877f, 3.302469253540039f, -0.054684557020664215f,
- 0.0016338867135345936f, -2.7690939903259277f, 2.5339231491088867f,
- 0.5221245884895325f, -0.17638295888900757f, -0.31949788331985474f,
- 2.323598623275757f, 0.2059188336133957f, -0.01752222701907158f,
- -1.1489571332931519f, 2.984546422958374f, -0.19020183384418488f,
- 0.010538210161030293f, -1.4752967357635498f, 1.8535363674163818f,
- 0.8166176080703735f, -0.5628044009208679f, -1.9997116327285767f,
- -2.098487377166748f, 0.3857003152370453f, -1.727634310722351f,
- -0.15402595698833466f, 3.4408957958221436f, -0.2683435082435608f,
- 0.023594189435243607f, 0.31177008152008057f, 3.039790391921997f,
- 0.4344753623008728f, 0.00961180217564106f, -0.45155903697013855f,
- -2.201700210571289f, -2.646042585372925f, 0.01605065166950226f,
- 0.13390818238258362f, -3.6872377395629883f, 0.7149689793586731f,
- -0.04385269060730934f, -0.10101474821567535f, 2.27207088470459f,
- 0.34321531653404236f, 0.04720620810985565f, -0.7443615198135376f,
- -5.059787750244141f, 0.02228701300919056f, -0.0019030789844691753f,
- 0.0005637781578116119f, 3.129986524581909f, 0.09729813039302826f,
- -0.024177810177206993f, -1.8108325004577637f, 3.3083548545837402f,
- -0.7547611594200134f, -0.44618573784828186f, -0.21931181848049164f,
- -2.2004897594451904f, -2.6458916664123535f, 0.09130378812551498f,
- 0.2665710747241974f, -0.5643004775047302f, -0.33926868438720703f,
- -0.6195669174194336f, 0.7023015022277832f, -5.160871982574463f,
- -0.0034803124144673347f, -0.011646432802081108f, 0.012036978267133236f,
- -3.1459012031555176f, -1.4449714422225952f, -0.0737914890050888f,
- 0.6170333623886108f, 2.4652867317199707f, -0.05448657274246216f,
- -0.16370250284671783f, -0.6195785999298096f, -4.647854328155518f,
- -0.00691827479749918f, 0.038473278284072876f, 0.032320134341716766f,
- -4.32373571395874f, 0.004058001562952995f, -0.004842042922973633f,
- -0.011608919128775597f, 2.67899489402771f, -0.21639853715896606f,
- -0.004101857542991638f, -2.9161124229431152f, 4.8421549797058105f,
- 0.04978768154978752f, -0.04711421951651573f, -0.06902165710926056f,
- -0.16757740080356598f, 0.021471548825502396f, -0.09624961763620377f,
- -0.05160382017493248f, -4.769743919372559f, 0.1121208518743515f,
- -0.024247704073786736f, -0.20753318071365356f, 0.47400805354118347f,
- 0.46464693546295166f, -2.335646390914917f, -0.00663331151008606f,
- -1.8362538814544678f, -1.3952279090881348f, -0.0224999338388443f,
- 0.9369595050811768f, -0.32786062359809875f, -1.2076143026351929f,
- -0.4175379276275635f, 0.920979917049408f
+ 0.4476310610771179f, -0.8712111711502075f, 0.1288183629512787f,
+ 0.792915940284729f, -0.37857282161712646f, -0.5999379754066467f,
+ -0.09074056148529053f, 0.7428027987480164f, -0.806922197341919f,
+ -0.06513206660747528f, 0.0015326066641137004f, 0.03463340178132057f,
+ -0.45006656646728516f, 0.8719941973686218f, -0.3037683367729187f,
+ -0.4873082637786865f, 0.3746185898780823f, -0.35810524225234985f,
+ 0.16044893860816956f, -0.5600897073745728f, 0.7878753542900085f,
+ -0.057914119213819504f, -0.05587376281619072f, -0.32516810297966003f,
+ -0.054352328181266785f, 0.7050285339355469f, 0.4860455393791199f,
+ -0.26461395621299744f, 0.4708733856678009f, 0.6615039110183716f,
+ -0.6717810034751892f, 0.25766628980636597f, -0.20877176523208618f,
+ 0.4233146607875824f, -0.8261363506317139f, 0.6218927502632141f,
+ -0.7742594480514526f, -0.09597370028495789f, -0.016565624624490738f,
+ 0.3243916630744934f, 0.730843722820282f, 0.3434459865093231f,
+ -0.6115328073501587f, 0.05172838270664215f, -0.8085055947303772f,
+ -0.23082391917705536f, 0.25912967324256897f, 0.48390132188796997f,
+ -0.24071218073368073f, -0.21271266043186188f, 0.15473750233650208f,
+ 1.1897283792495728f, -0.605708122253418f, 0.8830835223197937f,
+ -0.8744847774505615f, -0.10007809847593307f, 0.14527341723442078f,
+ -0.3217686712741852f, -0.6968852281570435f, -0.12154795229434967f,
+ -0.8508146405220032f, -0.23319056630134583f, 0.27581697702407837f,
+ 0.49674510955810547f, 0.2984752655029297f, -0.24128034710884094f,
+ -0.6872956156730652f, 0.6757507920265198f, -0.9962611198425293f,
+ -0.24671325087547302f, 0.04458463191986084f, 0.5092823505401611f,
+ -0.755521833896637f, 0.1556125432252884f, -0.19932909309864044f,
+ 0.8233707547187805f, -0.33803892135620117f, 0.1641087830066681f,
+ -0.1640886515378952f, 0.2807005047798157f, 0.11740092933177948f,
+ 0.9441748261451721f, -0.19026336073875427f, 0.1905575692653656f,
+ -0.7156860828399658f, -0.30425164103507996f, -0.6542183756828308f,
+ -0.048906467854976654f, 0.6393331289291382f, -0.5186219811439514f,
+ -0.15262047946453094f, 0.729802668094635f, 0.4545252323150635f,
+ -0.5133960843086243f, 0.26521825790405273f, -0.4991665482521057f,
+ -0.4927463233470917f, -0.07089114189147949f, 0.33401384949684143f,
+ 0.3023838400840759f, 0.7638855576515198f, -0.4981137812137604f,
+ -0.5541877150535583f, -0.11165370792150497f, 0.3648005425930023f,
+ 0.5878415107727051f, 0.11315485090017319f, -1.0285885334014893f,
+ -0.45433279871940613f, -0.7583989500999451f, -0.14650657773017883f,
+ 0.24613532423973083f, -0.2773364782333374f, -0.9591728448867798f,
+ 0.5280625224113464f, 0.5242233276367188f, -0.39398059248924255f,
+ -0.8652032613754272f, 0.5243239998817444f, 0.5806688666343689f,
+ 0.42181527614593506f, -0.09341983497142792f, 0.12693725526332855f,
+ -0.39904987812042236f, 0.15719516575336456f, 0.16289645433425903f,
+ 0.20234838128089905f, -0.25414201617240906f
};
static const float av1_hd_partition_breakout_nn_bias_64_layer0[32] = {
- -0.9125985503196716f, 0.9354565739631653f, -0.9121711850166321f,
- -1.1538583040237427f, 0.08968774974346161f, 1.1123112440109253f,
- 1.4973655939102173f, 0.608467698097229f, 0.18589675426483154f,
- -1.441516399383545f, 0.9556304812431335f, 0.7926673889160156f,
- -1.4284138679504395f, -0.7179064750671387f, -0.047664791345596313f,
- -1.2288035154342651f, 0.6012410521507263f, 0.18902179598808289f,
- -0.89959716796875f, 0.23802867531776428f, -1.300124168395996f,
- -0.08421888202428818f, 0.28658464550971985f, -0.27196547389030457f,
- -1.0962705612182617f, -0.350778728723526f, 0.8084302544593811f,
- -0.11797107756137848f, -0.9147244691848755f, -0.9423892498016357f,
- -0.46693262457847595f, -1.179872989654541f
+ -0.34709370136260986f, -0.009843609295785427f, -1.6221565008163452f,
+ -0.668915867805481f, 0.9886232018470764f, 0.6522868275642395f,
+ -0.37716832756996155f, -0.07447462528944016f, 0.3766818642616272f,
+ -1.1512787342071533f, 0.40794867277145386f, -0.14982998371124268f,
+ -0.19331441819667816f, -0.9201817512512207f, 0.48975181579589844f,
+ -0.137226402759552f, -0.5480761528015137f, -0.14310558140277863f,
+ -0.035678427666425705f, -0.724799633026123f, 0.6203687787055969f,
+ -0.26913538575172424f, -0.45061108469963074f, 1.1059123277664185f,
+ -0.9563385844230652f, 0.40938442945480347f, 0.4411890506744385f,
+ 0.1871682107448578f, 0.004259683191776276f, -0.050544340163469315f,
+ 0.789713442325592f, 0.26867425441741943f
};
static const float av1_hd_partition_breakout_nn_weights_64_layer1[32] = {
- 1.888397216796875f, -1.6998655796051025f, 2.5516481399536133f,
- 31.333364486694336f, -7.420342922210693f, -1.175126552581787f,
- -2.0703518390655518f, -1.0818297863006592f, -1.0498319864273071f,
- 12.22414779663086f, -0.6940112113952637f, -0.3731234669685364f,
- 11.562948226928711f, 0.6934157013893127f, 1.0499767065048218f,
- 15.071033477783203f, -2.1736562252044678f, 0.47170618176460266f,
- 1.8899933099746704f, 0.2540336847305298f, 27.493698120117188f,
- 0.8192774057388306f, -0.011624461971223354f, 1.5983129739761353f,
- 34.282840728759766f, -15.804471969604492f, 1.1542948484420776f,
- -0.09561372548341751f, 4.60494327545166f, 1.2926712036132812f,
- 0.4446795880794525f, -0.7026389241218567f
+ 0.7450170516967773f, 0.1674197018146515f, 8.603666305541992f,
+ -1.4572426080703735f, -0.6806990504264832f, -0.36511117219924927f,
+ -0.21973447501659393f, -0.33904269337654114f, 0.494196355342865f,
+ 2.428866386413574f, -0.33053985238075256f, 0.5259426236152649f,
+ 0.7364671230316162f, 1.8298554420471191f, -0.20149026811122894f,
+ 0.4657254219055176f, 0.9128966331481934f, 0.41023728251457214f,
+ 0.4559532403945923f, -0.49007904529571533f, -0.6186080574989319f,
+ 0.6019283533096313f, 0.41913628578186035f, -0.9652310609817505f,
+ -1.266809105873108f, -0.6605119109153748f, -1.0554698705673218f,
+ 0.28501376509666443f, 0.40640202164649963f, 0.4747793972492218f,
+ -0.3028396964073181f, -0.020081687718629837f
};
static const float av1_hd_partition_breakout_nn_bias_64_layer1[1] = {
- -0.15468640625476837f
+ -0.35164690017700195f
};
static const float
av1_hd_partition_breakout_nn_weights_32_layer0[FEATURE_SIZE * 32] = {
- -0.18351112306118011f, -0.15651831030845642f, 0.0013713851803913713f,
- 0.01054045557975769f, -3.780832529067993f, -0.15531419217586517f,
- 0.03428387641906738f, -0.027534646913409233f, -4.079058647155762f,
- 0.05176529288291931f, -0.0065496210008859634f, -0.08574667572975159f,
- -1.9617663621902466f, 0.3167760670185089f, -0.33277764916419983f,
- 0.7234531044960022f, 2.2539572715759277f, -0.02695995196700096f,
- 0.005501582287251949f, -3.854844093322754f, -2.5093464851379395f,
- -1.8817286491394043f, -0.13509777188301086f, 0.8389045000076294f,
- -0.8123356103897095f, -2.210233449935913f, 0.04169511795043945f,
- 1.3030999898910522f, 1.0332109928131104f, 0.15076060593128204f,
- -0.02417255938053131f, -3.10256290435791f, -3.5063133239746094f,
- 0.1668155938386917f, -0.11349064111709595f, -0.08673816174268723f,
- 0.2023429572582245f, 0.2143218219280243f, -2.2571842670440674f,
- 0.1890597939491272f, -2.0899853706359863f, -0.5377482771873474f,
- -0.24231456220149994f, 0.7975266575813293f, -1.104709267616272f,
- 0.42163029313087463f, -1.588227391242981f, -0.5404914021492004f,
- 0.39026254415512085f, 0.3967314660549164f, -2.0958690643310547f,
- -0.1487407088279724f, -1.2271692752838135f, -2.4342153072357178f,
- 0.0073483348824083805f, 0.4332350790500641f, 1.8399561643600464f,
- 0.20915934443473816f, 0.014152542687952518f, -2.279658079147339f,
- 2.1412830352783203f, -1.2423213720321655f, 0.10377844423055649f,
- 0.0026721248868852854f, -1.515641212463379f, 0.011470284312963486f,
- 0.433901309967041f, 0.03397373855113983f, 1.9424909353256226f,
- 0.007673368323594332f, -0.019963057711720467f, -3.2635231018066406f,
- -3.814363956451416f, -0.008298925124108791f, -0.0006046147318556905f,
- -0.03078455477952957f, -2.5249812602996826f, -1.5319974422454834f,
- 0.03523765504360199f, 0.47840240597724915f, -2.459254503250122f,
- -0.060729049146175385f, 0.3428601920604706f, 1.0676825046539307f,
- -3.196352481842041f, 0.00556157436221838f, -0.014416366815567017f,
- 0.00013636458606924862f, 0.5342910289764404f, 0.08967946469783783f,
- -1.079426884651184f, 0.0459836907684803f, 1.169906735420227f,
- 0.23663409054279327f, 0.15936784446239471f, 0.12904685735702515f,
- -1.941832423210144f, -2.3517515659332275f, -0.4566086530685425f,
- 1.001920223236084f, -2.479144334793091f, -1.4872301816940308f,
- 0.44298961758613586f, 0.0910615548491478f, 0.34078770875930786f,
- -1.0173975229263306f, 0.2069491446018219f, 0.8345462679862976f,
- 1.5491070747375488f, 0.20149222016334534f, 0.2451339066028595f,
- -1.176571011543274f, -2.3429925441741943f, 0.2398013025522232f,
- -0.9587415456771851f, -0.05250968039035797f, -3.1348395347595215f,
- -0.55987948179245f, -0.040084246546030045f, 0.29037174582481384f,
- 1.663692593574524f, 0.6011162400245667f, 0.014114673249423504f,
- -0.9607003331184387f, -1.0632480382919312f, -0.9293855428695679f,
- -0.7134267091751099f, 0.9321397542953491f
+ -0.8952129483222961f, -0.23612700402736664f, -0.022868705913424492f,
+ 0.504967451095581f, 0.3275434970855713f, 1.1059952974319458f,
+ -0.34120088815689087f, -0.07344819605350494f, 0.15775921940803528f,
+ -0.04879460483789444f, 0.1439337134361267f, 1.2064639329910278f,
+ 0.006475068163126707f, -0.4494408369064331f, -0.10751494765281677f,
+ 1.1100929975509644f, 0.12578976154327393f, 0.4349801242351532f,
+ 0.5385650396347046f, -0.4146132171154022f, -0.6666572690010071f,
+ 0.1782642900943756f, -0.43637481331825256f, -0.522347629070282f,
+ -0.7048850655555725f, -0.21371017396450043f, 0.1338515281677246f,
+ 0.9378389716148376f, -0.9088003635406494f, 0.3172213137149811f,
+ -0.6936320066452026f, 0.6413770318031311f, -0.6204543113708496f,
+ -0.20075663924217224f, 0.011177961714565754f, 0.3956983685493469f,
+ -0.23885823786258698f, 0.7718412280082703f, 0.013668450526893139f,
+ -0.11212749779224396f, -0.3808520436286926f, -0.019013067707419395f,
+ -0.5678153038024902f, 1.038125991821289f, -0.0636964738368988f,
+ -0.9305693507194519f, 0.5443329811096191f, 0.5650880932807922f,
+ -0.2772569954395294f, -0.7757272720336914f, 0.02383701130747795f,
+ 0.5687315464019775f, -0.4549035429954529f, -0.16180022060871124f,
+ 0.033420149236917496f, 0.5288522839546204f, -0.3871588110923767f,
+ 0.712287962436676f, 0.008839298039674759f, -0.4081243574619293f,
+ -0.44496551156044006f, -0.49737873673439026f, 0.5079053044319153f,
+ 0.4881437122821808f, 0.6699126362800598f, 0.28043776750564575f,
+ -0.5355777740478516f, 0.2881544232368469f, 0.33460190892219543f,
+ -0.26208120584487915f, -0.44822946190834045f, -0.733630895614624f,
+ -0.35417506098747253f, 0.6161319613456726f, -0.0993569940328598f,
+ -0.7897841334342957f, -0.551094651222229f, 0.4497978985309601f,
+ -0.31630903482437134f, 0.46718651056289673f, -0.19858941435813904f,
+ 1.0053157806396484f, -0.3194795548915863f, 0.31316739320755005f,
+ -0.035107459872961044f, -0.7919036746025085f, 0.04070507362484932f,
+ -0.325952410697937f, -0.8687034845352173f, -0.30160313844680786f,
+ 0.16643138229846954f, 1.1927493810653687f, -0.5958532094955444f,
+ -0.38102859258651733f, -0.06056114658713341f, 1.063319206237793f,
+ -0.028595328330993652f, 0.7656055688858032f, 0.5335075855255127f,
+ -0.6592297554016113f, -0.3134363293647766f, -0.8477844595909119f,
+ 0.14632871747016907f, 0.5351374745368958f, -0.007659543305635452f,
+ -0.8012065291404724f, 0.47477757930755615f, 0.4271889626979828f,
+ -0.6191681623458862f, -0.14256015419960022f, 0.038999997079372406f,
+ 0.4897593557834625f, -0.713534414768219f, 0.07782091945409775f,
+ -0.28364527225494385f, -0.17344607412815094f, -0.24871549010276794f,
+ 0.6229367852210999f, 0.35328975319862366f, -0.48995962738990784f,
+ 0.9903460741043091f, -0.1367904245853424f, -0.14807336032390594f,
+ -0.28647252917289734f, 0.6976669430732727f, -0.5536572933197021f,
+ 0.3724263310432434f, -0.7798140645027161f
};
static const float av1_hd_partition_breakout_nn_bias_32_layer0[32] = {
- -0.11753689497709274f, -0.7140882015228271f, -0.9722395539283752f,
- -0.6984077095985413f, -0.726676881313324f, -0.36886507272720337f,
- -0.17676463723182678f, -0.10097962617874146f, -0.29938116669654846f,
- -1.727449893951416f, 0.5804333090782166f, -1.2764358520507812f,
- -1.108852505683899f, -0.8272318840026855f, 0.6785213947296143f,
- 0.11738266795873642f, -2.057903528213501f, -0.37823784351348877f,
- -0.9595664143562317f, -1.0248923301696777f, -0.23907040059566498f,
- -0.8479264974594116f, -0.13894951343536377f, -1.7720012664794922f,
- -1.6527719497680664f, -0.20360442996025085f, -1.7053756713867188f,
- 0.8720151782035828f, -1.0570132732391357f, -0.7993026971817017f,
- 0.46637022495269775f, -0.8374565243721008f
+ 0.5906000733375549f, 0.8543934226036072f, 0.43217557668685913f,
+ -0.643294632434845f, 0.30037060379981995f, -0.7660125494003296f,
+ 0.1731051206588745f, -0.7480033040046692f, -0.6353603005409241f,
+ 0.4871520698070526f, 0.7321553230285645f, 0.2810375988483429f,
+ -0.43400588631629944f, -0.17055644094944f, -0.8143026232719421f,
+ 0.13951803743839264f, -1.2939090728759766f, -0.5045616626739502f,
+ -1.295732021331787f, -0.9267586469650269f, -0.13748344779014587f,
+ -0.8858642578125f, 0.24190591275691986f, 0.34041425585746765f,
+ 0.05495613068342209f, 0.25397151708602905f, 0.24138814210891724f,
+ 0.12242670357227325f, -0.5556198358535767f, -0.46377164125442505f,
+ 0.6218398213386536f, 0.7764495015144348f
};
static const float av1_hd_partition_breakout_nn_weights_32_layer1[32] = {
- -0.329619824886322f, 4.491082191467285f, 11.859853744506836f,
- -0.2909659743309021f, -20.884370803833008f, 1.2395148277282715f,
- 0.7091589570045471f, -2.458725690841675f, 1.714850664138794f,
- 2.3455934524536133f, 0.7239453792572021f, 11.537474632263184f,
- 3.062530755996704f, 2.7946059703826904f, -1.4265711307525635f,
- 0.44554367661476135f, -1.534838080406189f, -4.844552993774414f,
- 18.15884780883789f, 5.826704978942871f, 0.5580787062644958f,
- 11.919114112854004f, 0.24757426977157593f, 0.30521970987319946f,
- -1.6404640674591064f, 0.848618745803833f, -0.8315955996513367f,
- -0.8414434194564819f, 6.57717227935791f, 2.829108476638794f,
- -0.25260433554649353f, -0.4767799377441406f
+ 0.6254786252975464f, -0.724627673625946f, 0.5428725481033325f,
+ 0.8609197735786438f, -0.33933258056640625f, 0.6795636415481567f,
+ 0.5508725643157959f, 1.4960711002349854f, 0.8083735108375549f,
+ -0.3456595838069916f, 0.5514245629310608f, 0.28833192586898804f,
+ 0.7953433394432068f, -0.009960951283574104f, -1.7862937450408936f,
+ 0.1605624407529831f, 1.6885508298873901f, -0.49641919136047363f,
+ -4.162553310394287f, -2.510878086090088f, 0.3701621890068054f,
+ -0.5248422622680664f, 0.5657362937927246f, 0.48716917634010315f,
+ -0.23934587836265564f, 0.45801791548728943f, 0.3940700590610504f,
+ 0.07297442853450775f, 0.287155419588089f, -0.0904238149523735f,
+ -0.608720064163208f, -0.7513124942779541f
};
static const float av1_hd_partition_breakout_nn_bias_32_layer1[1] = {
- 0.04493425786495209f
+ 0.1117519736289978f
};
static const float
av1_hd_partition_breakout_nn_weights_16_layer0[FEATURE_SIZE * 32] = {
- -0.03770807012915611f, 0.1042923852801323f, 0.3589613735675812f,
- 0.8915500044822693f, -0.025765806436538696f, -0.8080071806907654f,
- 0.11935761570930481f, 1.081723690032959f, -1.0217446088790894f,
- -0.05894668400287628f, -0.0417763888835907f, 1.9492236375808716f,
- -1.6240522861480713f, -1.0665417909622192f, -0.4230113625526428f,
- 2.6151089668273926f, -1.0250322818756104f, -1.013023853302002f,
- -0.3417854607105255f, 1.2021280527114868f, -2.932366132736206f,
- -1.5600024461746216f, 0.013506319373846054f, 1.827286958694458f,
- 0.5162537097930908f, 0.37939924001693726f, -0.12543700635433197f,
- -3.9469857215881348f, -0.5308187007904053f, 0.055751170963048935f,
- -2.1356542110443115f, 0.9524611830711365f, -1.2815498113632202f,
- -0.2583577334880829f, 0.5526014566421509f, -0.991321861743927f,
- 0.4657832682132721f, -0.32342609763145447f, 0.012717160396277905f,
- -4.607941150665283f, -0.7786141633987427f, -0.7440438270568848f,
- -0.10956363379955292f, 1.9957225322723389f, -3.557837724685669f,
- -0.06580297648906708f, -0.019815972074866295f, 0.48607689142227173f,
- 0.9143388271331787f, 0.7489029765129089f, 0.23694343864917755f,
- -2.3731346130371094f, -2.455751895904541f, 0.02998301573097706f,
- -1.352641224861145f, 1.4107884168624878f, -2.7167775630950928f,
- -1.3195676803588867f, -0.0024603719357401133f, 1.568859338760376f,
- -1.368638277053833f, -0.6318024396896362f, -0.8460041284561157f,
- 0.7431659698486328f, -0.17625141143798828f, -0.25151070952415466f,
- 0.19983969628810883f, 2.0860755443573f, -3.904813289642334f,
- -0.38988929986953735f, -0.002895161509513855f, 0.714531660079956f,
- -3.0156004428863525f, -0.6497556567192078f, 0.32678696513175964f,
- 0.5911594033241272f, -1.1386737823486328f, -0.6372389793395996f,
- -0.029834266752004623f, 2.0517866611480713f, -1.0608230829238892f,
- -0.955141544342041f, -0.10072463750839233f, 1.5374855995178223f,
- -1.2410826683044434f, -2.169691562652588f, 0.003649030579254031f,
- 2.0521841049194336f, -0.9012426733970642f, 0.6359571218490601f,
- 0.10054442286491394f, 0.9355012774467468f, -2.433013439178467f,
- -1.3059970140457153f, -0.062273263931274414f, 1.2142263650894165f,
- 0.07180827856063843f, -0.02799983322620392f, 0.012213055975735188f,
- -0.15036505460739136f, -1.5302343368530273f, -0.9248730540275574f,
- -1.2439353466033936f, 1.6203926801681519f, -0.26789119839668274f,
- 0.29344648122787476f, -0.12080803513526917f, 2.095102548599243f,
- -3.6576807498931885f, 0.32156985998153687f, -0.00601181061938405f,
- 0.3735950291156769f, -1.2174468040466309f, -0.21403838694095612f,
- -2.081902503967285f, 1.0142937898635864f, -1.2942562103271484f,
- -2.4739151000976562f, 0.11531462520360947f, 1.7982516288757324f,
- -2.203162670135498f, 0.1353112757205963f, -1.742370843887329f,
- 0.44314906001091003f, -1.1483821868896484f, -0.6968573331832886f,
- -0.059433888643980026f, 0.9399102330207825f
+ 0.3427221179008484f, 0.06073892489075661f, -0.06882427632808685f,
+ 1.1336252689361572f, 0.097809799015522f, -0.6385823488235474f,
+ 0.43692731857299805f, -0.6029260754585266f, -0.13368098437786102f,
+ -0.6389026045799255f, 0.2193300873041153f, 0.6348177790641785f,
+ 0.22468054294586182f, 0.13958360254764557f, -0.05150606855750084f,
+ -0.28101637959480286f, -0.4977607727050781f, -0.3215360641479492f,
+ -0.16947278380393982f, 1.0890949964523315f, -1.02256441116333f,
+ 0.34789103269577026f, 0.34546756744384766f, -0.08185210078954697f,
+ 0.24066677689552307f, -1.2535117864608765f, 0.04398220032453537f,
+ 0.8025804162025452f, -0.2541569769382477f, 0.3501926362514496f,
+ -0.3649175763130188f, -0.2356642484664917f, 0.8068639039993286f,
+ -0.23699556291103363f, -0.2195892035961151f, -0.1547853797674179f,
+ -0.15287409722805023f, 0.4898677468299866f, -0.3860497772693634f,
+ 0.32813069224357605f, -0.7604209184646606f, -0.6611303091049194f,
+ 0.006376428063958883f, 1.0097582340240479f, 0.18538667261600494f,
+ 0.5587815642356873f, 0.30065974593162537f, -0.8898136615753174f,
+ 0.63029944896698f, 0.839385986328125f, -0.2562950849533081f,
+ -0.5478677749633789f, -0.7537129521369934f, 0.019417811185121536f,
+ 0.23349952697753906f, 0.07770441472530365f, 0.0019170257728546858f,
+ 0.003552661743015051f, 0.012630019336938858f, 0.03414732217788696f,
+ -0.04959012567996979f, -0.21843530237674713f, -0.5894723534584045f,
+ 0.21037738025188446f, -0.4222605526447296f, -0.5836210250854492f,
+ 0.011957126669585705f, 0.0031051053665578365f, -0.5127609968185425f,
+ -0.7208832502365112f, 0.05614267289638519f, 1.1000267267227173f,
+ -0.7031041383743286f, -0.20904500782489777f, 0.2827422320842743f,
+ 0.5169914960861206f, -0.03972400724887848f, -0.24878446757793427f,
+ -0.6129308938980103f, 0.3691819906234741f, -0.5902302861213684f,
+ -0.44936394691467285f, -0.20858828723430634f, -0.1926502138376236f,
+ 0.015391144901514053f, -0.05831986665725708f, 0.0012774848146364093f,
+ 0.9318875670433044f, -0.07232701778411865f, -0.07282842695713043f,
+ -0.3941495716571808f, 0.6986988186836243f, -0.6321771144866943f,
+ -0.1466921716928482f, -0.18214966356754303f, -0.2929930090904236f,
+ -0.048453543335199356f, 0.5302963852882385f, 0.3090183138847351f,
+ -0.7714007496833801f, -0.03153527155518532f, -0.28210026025772095f,
+ -0.0035391938872635365f, 1.1437654495239258f, 0.3694542348384857f,
+ -0.8283377885818481f, -0.06223354861140251f, 0.7067912817001343f,
+ -0.5538484454154968f, -0.3097091019153595f, 0.24007144570350647f,
+ 0.5142718553543091f, 0.5205507278442383f, -0.7828086018562317f,
+ 0.02196805737912655f, -0.5540323853492737f, -0.5541141629219055f,
+ -0.5407395958900452f, 0.83812016248703f, 0.5495181679725647f,
+ -0.36925947666168213f, -0.5186708569526672f, 0.035871513187885284f,
+ 0.9290533661842346f, 0.7525129914283752f, 0.16506646573543549f,
+ 0.015168293379247189f, -0.22260211408138275f
};
static const float av1_hd_partition_breakout_nn_bias_16_layer0[32] = {
- -2.500817060470581f, -1.1564264297485352f, 0.2977291941642761f,
- -0.255332350730896f, -0.44481080770492554f, -0.6241381168365479f,
- -0.4780181348323822f, -0.6444894075393677f, -1.2649517059326172f,
- -1.3098241090774536f, 0.10575855523347855f, -0.6572186946868896f,
- -0.13408327102661133f, -2.2591185569763184f, -0.6009128093719482f,
- -0.9144248962402344f, 0.5553908944129944f, -0.40132245421409607f,
- 0.3925309181213379f, 0.9891119003295898f, 0.7784745097160339f,
- -0.47138360142707825f, -1.2775310277938843f, -0.3515026867389679f,
- -0.4055931568145752f, -0.4989534318447113f, 0.7892708778381348f,
- -0.5805770754814148f, -1.247983694076538f, -0.46460986137390137f,
- -1.3504952192306519f, -0.29709282517433167f
+ 0.3051907420158386f, -0.7940329313278198f, 0.37281763553619385f,
+ -0.47349175810813904f, 0.9813013076782227f, -0.6225157976150513f,
+ -0.3041200339794159f, 0.18893377482891083f, -0.04121162369847298f,
+ 0.0799027606844902f, -0.06257568299770355f, 0.004709158092737198f,
+ 0.4165017306804657f, 0.40799248218536377f, -0.1605808585882187f,
+ 0.4690406322479248f, 0.2635003328323364f, 0.37693503499031067f,
+ 0.509822428226471f, 0.6932690739631653f, 0.3800438344478607f,
+ -0.9304105639457703f, 0.688030481338501f, -1.478393316268921f,
+ -0.08685281872749329f, -0.7200548648834229f, -0.9290581345558167f,
+ 0.47607457637786865f, -0.2131124883890152f, 0.4063669741153717f,
+ 0.6758500337600708f, 0.4055664539337158f
};
static const float av1_hd_partition_breakout_nn_weights_16_layer1[32] = {
- 0.37168681621551514f, 1.6819430589675903f, -0.24184460937976837f,
- -0.5973770022392273f, -0.9343476295471191f, 9.948596000671387f,
- -2.4088406562805176f, 0.7863813638687134f, -0.4910288453102112f,
- -8.380901336669922f, -0.41362130641937256f, 5.82503604888916f,
- -0.4502200186252594f, -1.826108694076538f, 4.183648109436035f,
- 2.714763879776001f, -0.26116943359375f, 2.803555727005005f,
- 0.8531390428543091f, 1.0828324556350708f, 0.8189658522605896f,
- -2.4576377868652344f, -0.3627878427505493f, 2.138772487640381f,
- -0.016609186306595802f, 0.7119042873382568f, 0.5726228356361389f,
- 1.508818507194519f, 7.78598690032959f, 2.531076669692993f,
- 6.370460510253906f, -0.04699355363845825f
+ 0.4050608277320862f, -0.41333451867103577f, 0.17236362397670746f,
+ 0.024612678214907646f, 0.9329894781112671f, -0.838448166847229f,
+ -1.4745935201644897f, 0.09989786148071289f, -0.10163675248622894f,
+ 0.17781522870063782f, 2.2620022296905518f, -0.3684936761856079f,
+ -0.6526914834976196f, 0.2942826449871063f, -0.35902100801467896f,
+ 0.10697637498378754f, 0.20386190712451935f, 1.2234491109848022f,
+ 0.30403849482536316f, 0.35678544640541077f, 0.29488465189933777f,
+ 1.736344814300537f, 0.5220925807952881f, -1.7380571365356445f,
+ -0.2642151713371277f, 0.7513654828071594f, -1.9644155502319336f,
+ 0.1906810700893402f, -0.35255730152130127f, 0.5716033577919006f,
+ 0.653854250907898f, -0.26972928643226624f
};
static const float av1_hd_partition_breakout_nn_bias_16_layer1[1] = {
- 0.42323046922683716f
+ 0.35929590463638306f
};
static const float
av1_hd_partition_breakout_nn_weights_8_layer0[FEATURE_SIZE * 32] = {
- -0.9726589918136597f, -0.5411639213562012f, 0.7087171673774719f,
- 1.7313759326934814f, -2.402184009552002f, -0.047024089843034744f,
- -0.027457909658551216f, 1.6469149589538574f, 0.20576165616512299f,
- -0.7170919179916382f, -0.10603219270706177f, 2.669275999069214f,
- -0.2676953077316284f, -0.16688808798789978f, -0.03148927167057991f,
- 1.4055346250534058f, -0.7267357110977173f, -1.3847031593322754f,
- 0.21190257370471954f, -1.012730598449707f, -1.2875311374664307f,
- -0.9335927963256836f, 0.9185367226600647f, -0.262455552816391f,
- -1.8908239603042603f, -0.7330380082130432f, 0.580338716506958f,
- 1.0240660905838013f, -0.741085946559906f, -0.4254375696182251f,
- 1.0939886569976807f, 0.8454716205596924f, 0.028208844363689423f,
- -1.108168363571167f, -0.5589184165000916f, 2.6788110733032227f,
- -2.6662967205047607f, -0.7947397232055664f, 0.0377255417406559f,
- 2.2486352920532227f, -0.6704809069633484f, 0.7823348641395569f,
- 0.18206557631492615f, 1.8088916540145874f, -0.2850566804409027f,
- -0.2228451520204544f, -0.0077590541914105415f, 0.267416387796402f,
- -0.11488392949104309f, -0.3818961977958679f, 0.15274959802627563f,
- 1.7191359996795654f, -0.1928352564573288f, 0.1540122777223587f,
- -1.7266273498535156f, 0.16492529213428497f, -0.26917898654937744f,
- 0.0453917570412159f, 0.006803665775805712f, -3.431412696838379f,
- -0.8123732209205627f, 0.20750433206558228f, 0.043349288403987885f,
- 1.5046347379684448f, -0.43123146891593933f, 0.496737003326416f,
- 0.13186758756637573f, 1.348464012145996f, -1.0381258726119995f,
- -0.14341726899147034f, -0.12307563424110413f, -0.47205984592437744f,
- -0.26413795351982117f, -0.4825311005115509f, 0.01020186860114336f,
- 1.8038893938064575f, 0.2135389894247055f, -0.5266213417053223f,
- -0.34630000591278076f, 2.1957969665527344f, -2.6997740268707275f,
- 0.22156886756420135f, 0.01723811775445938f, 0.9266343116760254f,
- -0.29436543583869934f, -0.21920204162597656f, 0.050585292279720306f,
- 1.8021506071090698f, 0.28632524609565735f, -1.0584778785705566f,
- -0.004406697116792202f, 1.6804509162902832f, 0.1515306681394577f,
- -0.5138083696365356f, -1.493039846420288f, 1.1335597038269043f,
- -2.868360757827759f, -0.7500830888748169f, -0.018952427431941032f,
- 1.7541661262512207f, -1.2600187063217163f, -0.19970472157001495f,
- -0.027286890894174576f, -2.7354300022125244f, -1.349392056465149f,
- -2.1341166496276855f, 0.1937991976737976f, -0.12994812428951263f,
- -0.8666549921035767f, 0.3629917502403259f, -0.2922564446926117f,
- 1.5711613893508911f, -1.5161727666854858f, -0.07508653402328491f,
- -0.015387722291052341f, 1.1435129642486572f, 0.08286841958761215f,
- -0.4614250361919403f, -0.26704689860343933f, 2.349350929260254f,
- -0.16499075293540955f, 0.025024881586432457f, 0.001498897559940815f,
- -0.04137025028467178f, -0.8304779529571533f, -0.20447227358818054f,
- 0.009074501693248749f, 0.8690717816352844f
+ -1.1174798011779785f, -0.02540789172053337f, -0.024517325684428215f,
+ 0.9283695220947266f, 0.514511227607727f, 0.008708865381777287f,
+ -0.7743955254554749f, 0.03391582891345024f, -0.4907812476158142f,
+ 0.06405021250247955f, 0.009607836604118347f, -0.06413879245519638f,
+ -0.3100389838218689f, 0.027276739478111267f, 0.02754838392138481f,
+ 0.2741911709308624f, -0.33674532175064087f, -0.12841999530792236f,
+ -0.07583238184452057f, 1.2658021450042725f, -0.07618967443704605f,
+ 0.018406564369797707f, 0.36800676584243774f, 1.1263890266418457f,
+ 0.29222336411476135f, -0.1937945932149887f, -0.781943678855896f,
+ 0.08040471374988556f, 0.06932314485311508f, 0.27592363953590393f,
+ 0.6615497469902039f, -0.3753996193408966f, 0.04914139211177826f,
+ -0.18584918975830078f, 0.3851499855518341f, -0.14877784252166748f,
+ 0.8386304974555969f, -0.9132646322250366f, 0.09750116616487503f,
+ -0.38142505288124084f, -0.27932387590408325f, -0.3158399760723114f,
+ 0.3219115734100342f, -0.11211645603179932f, -0.1601802557706833f,
+ 0.199840247631073f, -0.1438094526529312f, 1.4131516218185425f,
+ -0.6507886648178101f, -0.35280969738960266f, -0.07980803400278091f,
+ 0.9108555912971497f, -0.49617353081703186f, -0.2692374289035797f,
+ 0.019175099208950996f, 0.5736552476882935f, -0.48202967643737793f,
+ -0.26134830713272095f, 0.018342891708016396f, 0.6154197454452515f,
+ -0.18230408430099487f, -0.8209198713302612f, 0.6161503195762634f,
+ 0.4757719039916992f, 0.5258967876434326f, -0.9027984142303467f,
+ -0.47966063022613525f, 0.6204502582550049f, -1.342071294784546f,
+ 0.12434060871601105f, 0.14531831443309784f, -0.15101315081119537f,
+ -0.35774746537208557f, 0.08184701204299927f, -0.04071677848696709f,
+ 1.2944822311401367f, -0.06492389738559723f, 0.34913739562034607f,
+ -0.08534345775842667f, 1.3283460140228271f, -0.7555822134017944f,
+ 0.3319704234600067f, -0.17644117772579193f, 0.5248327255249023f,
+ -0.46480831503868103f, -0.23956067860126495f, 0.0013285222230479121f,
+ 0.5432708859443665f, 0.30651840567588806f, 0.37756049633026123f,
+ -0.03282826766371727f, 0.06274658441543579f, -0.2130001038312912f,
+ 0.2907213270664215f, 0.09125600755214691f, -0.133547842502594f,
+ -0.05953352898359299f, 0.032438505440950394f, 0.05902266874909401f,
+ -0.043947484344244f, -1.100358486175537f, 0.2626442313194275f,
+ 0.056724611669778824f, -0.19415289163589478f, 0.010464449413120747f,
+ -0.385423481464386f, -0.29392293095588684f, 0.5806636214256287f,
+ -0.28265658020973206f, 1.0120196342468262f, 0.17220932245254517f,
+ -0.9805957078933716f, -0.368004709482193f, -0.42056915163993835f,
+ -0.1005094051361084f, -0.740735650062561f, 0.8810465335845947f,
+ -0.4035398066043854f, 0.06286423653364182f, -0.26255688071250916f,
+ -0.24484986066818237f, -0.1917300969362259f, 0.23190940916538239f,
+ -0.5081801414489746f, -0.909449577331543f, 0.6952412128448486f,
+ -0.41929101943969727f, -0.28803330659866333f
};
static const float av1_hd_partition_breakout_nn_bias_8_layer0[32] = {
- 0.7538588047027588f, -0.0891113132238388f, 0.6207568049430847f,
- 0.1754276007413864f, -0.9016708731651306f, -0.5710541009902954f,
- 0.3960225582122803f, -0.5215008854866028f, 0.1935194879770279f,
- 0.1463070660829544f, -0.5508337020874023f, -0.25621265172958374f,
- -0.21759319305419922f, -0.6702320575714111f, -0.510668158531189f,
- -0.11305645108222961f, 0.9119726419448853f, 0.932460367679596f,
- 0.5708784461021423f, 0.9256153702735901f, 0.4665280282497406f,
- 0.2577301561832428f, 0.4606616199016571f, 0.4183802306652069f,
- -0.11034739017486572f, -1.2057222127914429f, -1.5402069091796875f,
- 0.36954033374786377f, -0.07251504063606262f, 0.4745430052280426f,
- -0.4834096133708954f, -0.08527179062366486f
+ 0.9311190843582153f, -0.04600553959608078f, 0.4736453592777252f,
+ -0.21461233496665955f, -0.6837071776390076f, 0.41968896985054016f,
+ 0.5667760372161865f, -0.295064777135849f, 0.6580685973167419f,
+ -0.12489219009876251f, 0.6441367864608765f, 0.017892301082611084f,
+ 0.3467213213443756f, 0.24851299822330475f, 0.27420252561569214f,
+ 0.6503739356994629f, -0.9539271593093872f, -0.7639755606651306f,
+ 0.3438030183315277f, -0.65008944272995f, 0.1381210833787918f,
+ 0.17897674441337585f, 0.6611759066581726f, -0.6359773278236389f,
+ 0.753626823425293f, -0.8921093940734863f, 0.7966955304145813f,
+ -0.20982062816619873f, 0.2511371076107025f, -0.23097757995128632f,
+ 1.1271780729293823f, 0.26889708638191223f
};
static const float av1_hd_partition_breakout_nn_weights_8_layer1[32] = {
- 1.1129406690597534f, -1.5291675329208374f, -0.8876321911811829f,
- 0.30224356055259705f, -0.40715256333351135f, -0.5691804885864258f,
- 0.8208304643630981f, -0.6657942533493042f, -1.1190624237060547f,
- 3.5144896507263184f, -0.4905086159706116f, -0.04449653998017311f,
- 0.8738682270050049f, 0.38138872385025024f, -2.4828171730041504f,
- -0.31892910599708557f, 0.4230459928512573f, 0.6047843098640442f,
- 0.5003713369369507f, 1.1065287590026855f, 0.97183758020401f,
- 0.2900541126728058f, 0.5317407846450806f, 0.6475094556808472f,
- 4.615662574768066f, -7.686334133148193f, -0.9662672877311707f,
- 0.6237889528274536f, -0.5734347701072693f, -0.4399879276752472f,
- -0.016914604231715202f, -0.6294510960578918f
+ 2.6572957038879395f, 0.17708347737789154f, 0.09978588670492172f,
+ -0.15976227819919586f, 2.394906997680664f, 0.2525644302368164f,
+ 0.4016534388065338f, -0.18184596300125122f, 0.20857582986354828f,
+ -0.6041266918182373f, 0.1736220121383667f, 0.5656698346138f,
+ 1.886559247970581f, 1.0891855955123901f, 0.3225744664669037f,
+ 0.5683131217956543f, -1.198813796043396f, -1.226926326751709f,
+ 0.42729490995407104f, -1.0998389720916748f, -1.6383575201034546f,
+ 0.47224658727645874f, 0.3989439904689789f, 0.003893804969266057f,
+ 0.2592338025569916f, -0.9896919131278992f, 0.23605197668075562f,
+ -1.2522234916687012f, 0.44355013966560364f, -0.37252721190452576f,
+ 0.4963497519493103f, -0.596142053604126f
};
static const float av1_hd_partition_breakout_nn_bias_8_layer1[1] = {
- 1.0166552066802979f
+ 0.8022159934043884f
};
static const float
@@ -3950,19 +3931,19 @@
{
// lowres
{
- 2.312415f, // p = 0.9099
- 1.848067f, // p = 0.8639
- 2.708050f, // p = 0.9375
- 1.567311f, // p = 0.8274
- 3.167687f, // p = 0.9596
+ 1.587045f, // p = 0.830200
+ 1.982054f, // p = 0.878900
+ 2.150334f, // p = 0.895700
+ 2.093809f, // p = 0.890300
+ 2.428837f, // p = 0.919000
},
// midres
{
- 2.312415f, // p = 0.9099
- 1.848067f, // p = 0.8639
- 2.708050f, // p = 0.9375
- 1.567311f, // p = 0.8274
- 3.167687f, // p = 0.9596
+ 2.044592f, // p = 0.885400
+ 1.961544f, // p = 0.876700
+ 2.940236f, // p = 0.949800
+ 2.301486f, // p = 0.909000
+ 2.265745f, // p = 0.906000
},
// hdres
{
@@ -3977,19 +3958,19 @@
{
// lowres
{
- 2.312415f, // p = 0.9099
- 1.848067f, // p = 0.8639
- 2.708050f, // p = 0.9375
- 1.567311f, // p = 0.8274
- 3.167687f, // p = 0.9596
+ 1.390049f, // p = 0.800600
+ 1.784893f, // p = 0.856300
+ 3.199090f, // p = 0.960800
+ 2.119666f, // p = 0.892800
+ 2.026000f, // p = 0.883500
},
// midres
{
- 2.312415f, // p = 0.9099
- 1.848067f, // p = 0.8639
- 2.708050f, // p = 0.9375
- 1.567311f, // p = 0.8274
- 3.167687f, // p = 0.9596
+ 1.587045f, // p = 0.830200
+ 1.982054f, // p = 0.878900
+ 2.150334f, // p = 0.895700
+ 2.093809f, // p = 0.890300
+ 2.428837f, // p = 0.919000
},
// hdres
{
@@ -4101,19 +4082,19 @@
{
// lowres
{
- -2.197225f, // p = 0.1
- -5.289093f, // p = 0.005021
- -2.522730f, // p = 0.07428
- -7.174085f, // p = 0.0007656
- -6.316499f, // p = 0.001803
+ -2.758650f, // p = 0.059600
+ -4.385780f, // p = 0.012300
+ -4.230259f, // p = 0.014340
+ -6.455747f, // p = 0.001569
+ -8.541803f, // p = 0.000195
},
// midres
{
- -2.197225f, // p = 0.1
- -5.289093f, // p = 0.005021
- -2.522730f, // p = 0.07428
- -7.174085f, // p = 0.0007656
- -6.316499f, // p = 0.001803
+ -4.881569f, // p = 0.007528
+ -4.656635f, // p = 0.009409
+ -3.037505f, // p = 0.045760
+ -6.679364f, // p = 0.001255
+ -5.944256f, // p = 0.002614
},
// hdres
{
@@ -4128,19 +4109,19 @@
{
// lowres
{
- -2.197225f, // p = 0.1
- -5.289093f, // p = 0.005021
- -2.522730f, // p = 0.07428
- -7.174085f, // p = 0.0007656
- -6.316499f, // p = 0.001803
+ -4.381672f, // p = 0.012350
+ -4.034317f, // p = 0.017390
+ -3.986288f, // p = 0.018230
+ -7.325495f, // p = 0.000658
+ -7.908147f, // p = 0.000368
},
// midres
{
- -2.197225f, // p = 0.1
- -5.289093f, // p = 0.005021
- -2.522730f, // p = 0.07428
- -7.174085f, // p = 0.0007656
- -6.316499f, // p = 0.001803
+ -2.758650f, // p = 0.059600
+ -4.385780f, // p = 0.012300
+ -4.230259f, // p = 0.014340
+ -6.455747f, // p = 0.001569
+ -8.541803f, // p = 0.000195
},
// hdres
{
diff --git a/av1/encoder/partition_strategy.c b/av1/encoder/partition_strategy.c
index a4a1f82..1c0f1bd 100644
--- a/av1/encoder/partition_strategy.c
+++ b/av1/encoder/partition_strategy.c
@@ -1588,16 +1588,29 @@
float rate_f = (float)AOMMIN(rd_stats->rate, INT_MAX);
rate_f = ((float)x->rdmult / 128.0f / 512.0f / (float)(1 << num_pels_log2)) *
rate_f;
- features[feature_index++] = rate_f;
+ features[feature_index++] =
+ cpi->sf.part_sf.ml_partition_search_breakout_model_index
+ ? log1pf((float)rate_f)
+ : rate_f;
const float dist_f =
(float)(AOMMIN(rd_stats->dist, INT_MAX) >> num_pels_log2);
- features[feature_index++] = dist_f;
+ features[feature_index++] =
+ cpi->sf.part_sf.ml_partition_search_breakout_model_index
+ ? log1pf((float)dist_f)
+ : dist_f;
- features[feature_index++] = (float)pb_source_variance;
+ features[feature_index++] =
+ cpi->sf.part_sf.ml_partition_search_breakout_model_index
+ ? log1pf((float)pb_source_variance)
+ : (float)pb_source_variance;
const int dc_q = (int)x->plane[0].dequant_QTX[0] >> (bit_depth - 8);
- features[feature_index++] = (float)(dc_q * dc_q) / 256.0f;
+ features[feature_index++] =
+ cpi->sf.part_sf.ml_partition_search_breakout_model_index
+ ? log1pf((float)(dc_q * dc_q) / 256.0f)
+ : (float)(dc_q * dc_q) / 256.0f;
+
assert(feature_index == FEATURES);
if (cpi->sf.part_sf.ml_partition_search_breakout_model_index) {
diff --git a/av1/encoder/speed_features.c b/av1/encoder/speed_features.c
index 56832c6..8cc56a2 100644
--- a/av1/encoder/speed_features.c
+++ b/av1/encoder/speed_features.c
@@ -233,12 +233,11 @@
}
if (is_720p_or_larger) {
- sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.9999999f;
- sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.9999999f;
- sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.9618367258814811f;
- sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.9990705139233304f;
- sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.9648891196441841f;
-
+ sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.5f;
+ sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.5042595622791082f;
+ sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.5f;
+ sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.8378425823517456f;
+ sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.8047585616503903f;
sf->part_sf.ml_partition_search_breakout_model_index = 1;
} else {
sf->part_sf.ml_partition_search_breakout_thresh[0] = -1.0f;
@@ -259,11 +258,11 @@
}
if (is_720p_or_larger) {
- sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.9583713938680828f;
- sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.9999999f;
- sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.9634239069901543f;
- sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.9000000000000001f;
- sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.9196596355880025f;
+ sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.5f;
+ sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.5042595622791082f;
+ sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.5f;
+ sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.8378425823517456f;
+ sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.8047585616503903f;
sf->part_sf.ml_partition_search_breakout_model_index = 1;
}
@@ -785,11 +784,11 @@
}
if (is_720p_or_larger) {
- sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.9999999f;
- sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.9999999f;
- sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.9618367258814811f;
- sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.9990705139233304f;
- sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.9648891196441841f;
+ sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.5f;
+ sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.5042595622791082f;
+ sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.5f;
+ sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.8378425823517456f;
+ sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.8047585616503903f;
sf->part_sf.ml_partition_search_breakout_model_index = 1;
} else {
sf->part_sf.ml_partition_search_breakout_thresh[0] = -1.0f;
@@ -812,11 +811,11 @@
}
if (is_720p_or_larger) {
- sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.9583713938680828f;
- sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.9999999f;
- sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.9634239069901543f;
- sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.9000000000000001f;
- sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.9196596355880025f;
+ sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.5f;
+ sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.5042595622791082f;
+ sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.5f;
+ sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.8378425823517456f;
+ sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.8047585616503903f;
sf->part_sf.ml_partition_search_breakout_model_index = 1;
}
