From 49fffb36710086e0a0540d194fd53e7954fd563c Mon Sep 17 00:00:00 2001 From: Brandon Amos Date: Fri, 3 Jun 2016 19:04:40 -0400 Subject: [PATCH] Correct inception tower sizes for #142. --- models/openface/nn2.def.lua | 20 ++++++++++---------- models/openface/nn4.def.lua | 20 ++++++++++---------- models/openface/nn4.small1.def.lua | 16 ++++++++-------- models/openface/nn4.small2.def.lua | 14 +++++++------- 4 files changed, 35 insertions(+), 35 deletions(-) diff --git a/models/openface/nn2.def.lua b/models/openface/nn2.def.lua index 66aeb0b..728049f 100644 --- a/models/openface/nn2.def.lua +++ b/models/openface/nn2.def.lua @@ -63,7 +63,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {128, 32}, reduceSize = {96, 16, 32, 64}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 1, 1, 1, 1), batchNorm = true }) @@ -74,7 +74,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {128, 64}, reduceSize = {96, 32, 64, 64}, - pool = nn.SpatialLPPooling(256, 2, 3, 3), + pool = nn.SpatialLPPooling(256, 2, 3, 3, 1, 1), batchNorm = true }) @@ -85,7 +85,7 @@ function createModel() kernelStride = {2, 2}, outputSize = {256, 64}, reduceSize = {128, 32, nil, nil}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 2, 2, 1, 1), batchNorm = true }) @@ -96,7 +96,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {192, 64}, reduceSize = {96, 32, 128, 256}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -107,7 +107,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {224, 64}, reduceSize = {112, 32, 128, 224}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -118,7 +118,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {256, 64}, reduceSize = {128, 32, 128, 192}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -129,7 +129,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {288, 64}, reduceSize = {144, 32, 128, 160}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -140,7 +140,7 @@ function createModel() kernelStride = {2, 2}, outputSize = {256, 128}, reduceSize = {160, 64, nil, nil}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 2, 2, 1, 1), batchNorm = true }) @@ -151,7 +151,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {384, 128}, reduceSize = {192, 48, 128, 384}, - pool = nn.SpatialLPPooling(960, 2, 3, 3), + pool = nn.SpatialLPPooling(960, 2, 3, 3, 1, 1), batchNorm = true }) @@ -162,7 +162,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {384, 128}, reduceSize = {192, 48, 128, 384}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 1, 1, 1, 1), batchNorm = true }) diff --git a/models/openface/nn4.def.lua b/models/openface/nn4.def.lua index dc3610b..2c31c4e 100644 --- a/models/openface/nn4.def.lua +++ b/models/openface/nn4.def.lua @@ -63,7 +63,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {128, 32}, reduceSize = {96, 16, 32, 64}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 1, 1, 1, 1), batchNorm = true }) @@ -74,7 +74,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {128, 64}, reduceSize = {96, 32, 64, 64}, - pool = nn.SpatialLPPooling(256, 2, 3, 3), + pool = nn.SpatialLPPooling(256, 2, 3, 3, 1, 1), batchNorm = true }) @@ -85,7 +85,7 @@ function createModel() kernelStride = {2, 2}, outputSize = {256, 64}, reduceSize = {128, 32, nil, nil}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 2, 2, 1, 1), batchNorm = true }) @@ -96,7 +96,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {192, 64}, reduceSize = {96, 32, 128, 256}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -107,7 +107,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {224, 64}, reduceSize = {112, 32, 128, 224}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -118,7 +118,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {256, 64}, reduceSize = {128, 32, 128, 192}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -129,7 +129,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {288, 64}, reduceSize = {144, 32, 128, 160}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -140,7 +140,7 @@ function createModel() kernelStride = {2, 2}, outputSize = {256, 128}, reduceSize = {160, 64, nil, nil}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 2, 2, 1, 1), batchNorm = true }) @@ -151,7 +151,7 @@ function createModel() kernelStride = {1}, outputSize = {384}, reduceSize = {192, 128, 384}, - pool = nn.SpatialLPPooling(960, 2, 3, 3), + pool = nn.SpatialLPPooling(960, 2, 3, 3, 1, 1), batchNorm = true }) @@ -162,7 +162,7 @@ function createModel() kernelStride = {1}, outputSize = {384}, reduceSize = {192, 128, 384}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 1, 1, 1, 1), batchNorm = true }) diff --git a/models/openface/nn4.small1.def.lua b/models/openface/nn4.small1.def.lua index 58c565e..d8327a4 100644 --- a/models/openface/nn4.small1.def.lua +++ b/models/openface/nn4.small1.def.lua @@ -63,7 +63,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {128, 32}, reduceSize = {96, 16, 32, 64}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 1, 1, 1, 1), batchNorm = true }) @@ -74,7 +74,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {128, 64}, reduceSize = {96, 32, 64, 64}, - pool = nn.SpatialLPPooling(256, 2, 3, 3), + pool = nn.SpatialLPPooling(256, 2, 3, 3, 1, 1), batchNorm = true }) @@ -85,7 +85,7 @@ function createModel() kernelStride = {2, 2}, outputSize = {256, 64}, reduceSize = {128, 32, nil, nil}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 2, 2, 1, 1), batchNorm = true }) @@ -96,7 +96,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {192, 64}, reduceSize = {96, 32, 128, 256}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -107,7 +107,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {224, 64}, reduceSize = {112, 32, 128, 224}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -118,7 +118,7 @@ function createModel() kernelStride = {2, 2}, outputSize = {256, 128}, reduceSize = {160, 64, nil, nil}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 2, 2, 1, 1), batchNorm = true }) @@ -129,7 +129,7 @@ function createModel() kernelStride = {1}, outputSize = {384}, reduceSize = {192, 128, 384}, - pool = nn.SpatialLPPooling(960, 2, 3, 3), + pool = nn.SpatialLPPooling(960, 2, 3, 3, 1, 1), batchNorm = true }) @@ -140,7 +140,7 @@ function createModel() kernelStride = {1}, outputSize = {384}, reduceSize = {192, 128, 384}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 1, 1, 1, 1), batchNorm = true }) diff --git a/models/openface/nn4.small2.def.lua b/models/openface/nn4.small2.def.lua index a110670..5a6fec7 100644 --- a/models/openface/nn4.small2.def.lua +++ b/models/openface/nn4.small2.def.lua @@ -64,7 +64,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {128, 32}, reduceSize = {96, 16, 32, 64}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 1, 1, 1, 1), batchNorm = true }) @@ -75,7 +75,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {128, 64}, reduceSize = {96, 32, 64, 64}, - pool = nn.SpatialLPPooling(256, 2, 3, 3), + pool = nn.SpatialLPPooling(256, 2, 3, 3, 1, 1), batchNorm = true }) @@ -86,7 +86,7 @@ function createModel() kernelStride = {2, 2}, outputSize = {256, 64}, reduceSize = {128, 32, nil, nil}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 2, 2, 1, 1), batchNorm = true }) @@ -97,7 +97,7 @@ function createModel() kernelStride = {1, 1}, outputSize = {192, 64}, reduceSize = {96, 32, 128, 256}, - pool = nn.SpatialLPPooling(640, 2, 3, 3), + pool = nn.SpatialLPPooling(640, 2, 3, 3, 1, 1), batchNorm = true }) @@ -108,7 +108,7 @@ function createModel() kernelStride = {2, 2}, outputSize = {256, 128}, reduceSize = {160, 64, nil, nil}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 2, 2, 1, 1), batchNorm = true }) @@ -119,7 +119,7 @@ function createModel() kernelStride = {1}, outputSize = {384}, reduceSize = {96, 96, 256}, - pool = nn.SpatialLPPooling(960, 2, 3, 3), + pool = nn.SpatialLPPooling(960, 2, 3, 3, 1, 1), batchNorm = true }) -- net:add(nn.Reshape(736,3,3)) @@ -131,7 +131,7 @@ function createModel() kernelStride = {1}, outputSize = {384}, reduceSize = {96, 96, 256}, - pool = nn.SpatialMaxPooling(3, 3, 2, 2), + pool = nn.SpatialMaxPooling(3, 3, 1, 1, 1, 1), batchNorm = true })