Fixes missing val image preprocess call

batra-mlp-lab · Nov 21, 2017 · 8e75892 · 8e75892
1 parent 9f9f442
commit 8e75892
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 0 deletions.
diff --git a/data/prepro_utils.lua b/data/prepro_utils.lua
@@ -74,6 +74,7 @@ function extractFeatures(model, opt, ndims, preprocessFn)
         ims = torch.DoubleTensor(r - i + 1, 3, opt.imgSize, opt.imgSize)
         for j = 1, r - i + 1 do
             ims[j] = loadImage(valList[i + j - 1], opt.imgSize)
+            ims[j] = preprocessFn(ims[j])
         end
         if opt.gpuid >= 0 then
             ims = ims:cuda()

diff --git a/scripts/download_model.sh b/scripts/download_model.sh