small bugfix thanks to @WendyShang

soumith · Jan 11, 2016 · 8cadc46 · 8cadc46
1 parent 9591b88
commit 8cadc46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/donkey.lua b/donkey.lua
@@ -119,7 +119,7 @@ testHook = function(self, path)
    local iH = input:size(2)
    local w1 = math.ceil((iW-oW)/2)
    local h1 = math.ceil((iH-oH)/2)
-   local out = image.crop(input, w1, h1, w1+oW, h1+oW) -- center patch
+   local out = image.crop(input, w1, h1, w1+oW, h1+oH) -- center patch
    -- mean/std
    for i=1,3 do -- channels
       if mean then out[{{i},{},{}}]:add(-mean[i]) end