search

lufficc / SSD

High quality, fast, modular reference implementation of SSD in PyTorch
MIT License
1.52k stars 384 forks source link

Error while using torchvision==0.10.0 #206

Open Sciroccogti opened 3 years ago

Sciroccogti commented 3 years ago

Maybe this line which detecting the version of torchvision is wrong:

https://github.com/lufficc/SSD/blob/68dc0a20efaf3997e58b616afaaaa21bf8ca3c05/ssd/utils/nms.py#L7

in fact, str '0.10.0' < '0.3.0'