TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

AvgPool shim

Browse Source
This commit is contained in:
Mike J Innes 2017-03-06 17:21:35 +00:00
parent 5d919175fc
commit 2a57150bce
1 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/layers/shims.jl
View File

@ -1,4 +1,4 @@
export Conv2D, MaxPool, Reshape export Conv2D, MaxPool, AvgPool, Reshape
type Conv2D <: Model type Conv2D <: Model
filter::Param{Array{Float64,4}} # [height, width, inchans, outchans] filter::Param{Array{Float64,4}} # [height, width, inchans, outchans]
@ -14,21 +14,25 @@ shape(c::Conv2D, in::Dims{2}) =
shape(c::Conv2D, in::Dims{3}) = shape(c::Conv2D, in::Dims{3}) =
shape(c, (in[1],in[2])) shape(c, (in[1],in[2]))
type MaxPool <: Model for Pool in :[MaxPool, AvgPool].args
@eval begin
type $Pool <: Model
size::Dims{2} size::Dims{2}
stride::Dims{2} stride::Dims{2}
end end
MaxPool(size; stride = (1,1)) = $Pool(size; stride = (1,1)) =
MaxPool(size, stride) $Pool(size, stride)
shape(c::MaxPool, in::Dims{2}) = shape(c::$Pool, in::Dims{2}) =
map(i -> (in[i]-c.size[i])÷c.stride[i]+1, (1,2)) map(i -> (in[i]-c.size[i])÷c.stride[i]+1, (1,2))
shape(c::MaxPool, in::Dims{3}) = shape(c::$Pool, in::Dims{3}) =
(shape(c, (in[1],in[2]))..., in[3]) (shape(c, (in[1],in[2]))..., in[3])
shape(c::MaxPool, in) = throw(ShapeError(c, in)) shape(c::$Pool) = nothing
end
end
immutable Reshape{N} immutable Reshape{N}
dims::Dims{N} dims::Dims{N}