Solved https://github.com/zhou13/lcnn/issues/53#issue-1005072261

zhou13 / lcnn

LCNN: End-to-End Wireframe Parsing

MIT License

494 stars 94 forks source link

Closed ibrahim-601 closed 3 years ago

ibrahim-601 commented 3 years ago

In PyTorch 1.4.x (index/128) at https://github.com/zhou13/lcnn/blob/88f281ab5421d51a62f1f84f97fea05afbf0c8d8/lcnn/models/line_vectorizer.py#L174 outputs integer values but in PyTorch 1.8.x it outputs float values. Updating it with (index//128) outputs integer values in both PyTorch versions and solves https://github.com/zhou13/lcnn/issues/53#issue-1005072261.

zhou13 commented 3 years ago

Good catch! Thanks for the contribution.