search

rajeevyasarla / UMSN-Face-Deblurring

Deblurring Face Images using Uncertainty Guided Multi-Stream Semantic Networks
MIT License
118 stars 34 forks source link

unable to test #12

Closed geekodour closed 4 years ago

geekodour commented 4 years ago

I installed the dependencies, but am not able to run. This is the log:

Traceback (most recent call last):
  File "test_face_deblur.py", line 13, in <module>
    import torchvision.utils as vutils
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torchvision/__init__.py", line 1, in <module>
    from torchvision import models
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torchvision/models/__init__.py", line 12, in <module>
    from . import detection
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torchvision/models/detection/__init__.py", line 1, in <module>
    from .faster_rcnn import *
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torchvision/models/detection/faster_rcnn.py", line 13, in <module>
    from .rpn import AnchorGenerator, RPNHead, RegionProposalNetwork
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torchvision/models/detection/rpn.py", line 8, in <module>
    from . import _utils as det_utils
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torchvision/models/detection/_utils.py", line 74, in <module>
    @torch.jit.script
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/__init__.py", line 364, in script
    graph = _script_graph(fn, _frames_up=_frames_up + 1)
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/__init__.py", line 359, in _script_graph
    ast = get_jit_ast(fn)
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 132, in get_jit_ast
    return build_def(SourceRangeFactory(source), py_ast.body[0])
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 151, in build_def
    build_stmts(ctx, body))
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 123, in build_stmts
    stmts = [build_stmt(ctx, s) for s in stmts]
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 123, in <listcomp>
    stmts = [build_stmt(ctx, s) for s in stmts]
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 140, in __call__
    return method(ctx, node)
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 205, in build_Assign
    rhs = build_expr(ctx, stmt.value)
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 140, in __call__
    return method(ctx, node)
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 314, in build_Call
    func = build_expr(ctx, expr.func)
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 140, in __call__
    return method(ctx, node)
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 300, in build_Attribute
    value = build_expr(ctx, expr.value)
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 140, in __call__
    return method(ctx, node)
  File "/home/geekodour/.virtualenvs/h3/lib/python3.7/site-packages/torch/jit/frontend.py", line 422, in build_Subscript
    raise NotSupportedError(base.range(), "slicing multiple dimensions at the same time isn't supported yet")
torch.jit.frontend.NotSupportedError: slicing multiple dimensions at the same time isn't supported yet
        proposals (Tensor): boxes to be encoded
    """

    # perform some unpacking to make it JIT-fusion friendly
    wx = weights[0]
    wy = weights[1]
    ww = weights[2]
    wh = weights[3]

    proposals_x1 = proposals[:, 0].unsqueeze(1)
                   ~~~~~~~~~ <--- HERE
    proposals_y1 = proposals[:, 1].unsqueeze(1)
    proposals_x2 = proposals[:, 2].unsqueeze(1)
    proposals_y2 = proposals[:, 3].unsqueeze(1)

    reference_boxes_x1 = reference_boxes[:, 0].unsqueeze(1)
    reference_boxes_y1 = reference_boxes[:, 1].unsqueeze(1)
    reference_boxes_x2 = reference_boxes[:, 2].unsqueeze(1)
    reference_boxes_y2 = reference_boxes[:, 3].unsqueeze(1)

would be great help if you can update the documentation to something that'll work directly or make a Dockerfile or something.

mowshon commented 4 years ago

Your torchvision are not compatible with torch==0.4.1. I made a pull request to solve this: https://github.com/rajeevyasarla/UMSN-Face-Deblurring/pull/13

Just create file requirements.txt:

numpy==1.18.0
opencv-python==4.1.2.30
Pillow==6.2.1
six==1.13.0
torch==0.4.1.post2
torchvision==0.2.1

And run

pip3 install -r requirements.txt