facebookresearch / deepmask

Torch implementation of DeepMask and SharpMask

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Crash for grayscale images

kmerkelbach opened this issue · comments

Hi,

I noticed that SharpMask crashes for grayscale images.
Out of the COCO-Stuff images, it crashed for the following files:

  • COCO_train2014_000000000821.jpg
  • COCO_train2014_000000030349.jpg
  • COCO_train2014_000000126531.jpg
  • COCO_train2014_000000134918.jpg
  • COCO_train2014_000000140092.jpg
  • COCO_train2014_000000155083.jpg
  • COCO_train2014_000000176397.jpg
  • COCO_train2014_000000207339.jpg
  • COCO_train2014_000000211867.jpg
  • COCO_train2014_000000296884.jpg
  • COCO_train2014_000000341892.jpg
  • COCO_train2014_000000349069.jpg
  • COCO_train2014_000000361516.jpg
  • COCO_train2014_000000369966.jpg
  • COCO_train2014_000000384910.jpg
  • COCO_train2014_000000389984.jpg
  • COCO_train2014_000000410498.jpg
  • COCO_train2014_000000421613.jpg
  • COCO_train2014_000000434837.jpg
  • COCO_train2014_000000445845.jpg
  • COCO_train2014_000000454000.jpg
  • COCO_train2014_000000505962.jpg
  • COCO_train2014_000000555583.jpg
    All of these files are grayscale.

The stack trace is as follows:
~/src/sharpmask/InferSharpMask.lua:96: bad argument #1 to 'copy' (sizes do not match at /tmp/luarocks_cutorch-scm-1-547/cutorch/lib/THC/THCTensorCopy.cu:31) stack traceback: [C]: in function 'copy' /work2/merkelbach/src/sharpmask/InferSharpMask.lua:96: in function 'forward' ...elbach/src/sharpmask/computeProposalsMerkelbach_test.lua:131: in main chunk [C]: in function 'dofile' .../src/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk [C]: at 0x00406670