Issue with pytorch version

[ibrahim-601]

I tried following the instructions on the repository. Pytorch 1.8.1 is installed with the instructions. But with that environment, the right points of lines get a little bit downward. But if I try the same with PyTorch 1.4.0 there is no issue. Can you please explain why is this happening? Or can you update the instructions for the correct PyTorch version? I have attached both outputs. The first one is with PyTorch 1.4.0 and the second one is with PyTorch 1.8.1.

PyTorch 1.4.0 sample:

PyTorch 1.8.1 sample:

[zhou13]

Hi @ibrahim-601, Thank you for reporting. This is an interesting finding. Could you reproduce the result using demo.py with the pre-trained model? Could you also upload the original image in PNG so I can investigate? If you can help to debug, could you help print/plot jmap and joff at

lcnn/lcnn/models/line_vectorizer.py

Line 162 in 88f281a

max_K = M.n_dyn_junc // n_type

and see if they are the same in Python 1.8.1 and Python 1.4.0?

[ibrahim-601]

Hi @zhou13, thank you very much for replying. The above result is produced with demo.py with the pre-trained model. I saved the jmap and joff values in a JSON file using below code

js_data = {'jmap_after_nms': jmap.cpu().numpy().tolist(), 'joff': joff.cpu().numpy().tolist()}
json.dump(js_data, open('jmap_joff.json','w'))

at

lcnn/lcnn/models/line_vectorizer.py

Line 162 in 88f281a

max_K = M.n_dyn_junc // n_type

The values of jmap and joff in PyTorch 1.4.0 and PyTorch 1.8.1 are the same.
I have uploaded the original image and some test results along with the JSON files here.