OrgLion-ML
/
openface
mirror of https://github.com/cmusatyalab/openface.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Correct inception tower sizes for #142.

This commit is contained in:
Brandon Amos 2016-06-03 19:04:40 -04:00
parent 130edcecc9
commit 49fffb3671
4 changed files with 35 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
models/openface/nn2.def.lua
View File

@ -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
})

20
models/openface/nn4.def.lua
View File

@ -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
})

16
models/openface/nn4.small1.def.lua
View File

@ -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
})

14
models/openface/nn4.small2.def.lua
View File

@ -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
})