schirrmacher

/

ormbg

like 49

With the default example it's working but not with my own images:

python ormbg/inference.py --image=examples/image/output-onlinepngtools.png 

/opt/conda/envs/ormbg/lib/python3.8/site-packages/torch/nn/_reduction.py:42: UserWarning: size_average and reduce args will be deprecated, please use reduction='mean' instead.
  warnings.warn(warning.format(ret))
Traceback (most recent call last):
  File "ormbg/inference.py", line 110, in <module>
    inference(parse_args())
  File "ormbg/inference.py", line 83, in inference
    result = net(image)
  File "/opt/conda/envs/ormbg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/opt/conda/envs/ormbg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/jovyan/ormbg/ormbg/models/ormbg.py", line 416, in forward
    hxin = self.conv_in(hx)
  File "/opt/conda/envs/ormbg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/opt/conda/envs/ormbg/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
    return forward_call(*args, **kwargs)
  File "/opt/conda/envs/ormbg/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 460, in forward
    return self._conv_forward(input, self.weight, self.bias)
  File "/opt/conda/envs/ormbg/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 456, in _conv_forward
    return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Given groups=1, weight of size [64, 3, 3, 3], expected input[1, 4, 1024, 1024] to have 3 channels, but got 4 channels instead