Wrong check of torchvision version in detr.util.misc.py

microsoft / table-transformer

Table Transformer (TATR) is a deep learning model for extracting tables from unstructured documents (PDFs and images). This is also the official repository for the PubTables-1M dataset and GriTS evaluation metric.

MIT License

2.31k stars 257 forks source link

Wrong check of torchvision version in detr.util.misc.py #26

Closed dantp-ai closed 2 years ago

dantp-ai commented 2 years ago

The version check here fails for torchvision.__version__ == '0.11.2' because it is wrongly comparing 0.1 < 0.7. Instead the comparison should be 0.11 < 0.7.

Note that a correct implementation should also consider cases like 0.4.2.post2.

bsmock commented 2 years ago

Thank you, we have now adopted the fix from the original DETR codebase. It will be in the next pull request.