Closed sieukim closed 7 months ago
The current torchvision.__version__ is 0.11.0, but '0.3.0' is larger than '0.11.0' when comparing by string. This makes an error in ssd/utils/nms.py. So I changed the version comparison code.
torchvision.__version__
'0.3.0'
'0.11.0'
ssd/utils/nms.py
The current
torchvision.__version__
is 0.11.0, but'0.3.0'
is larger than'0.11.0'
when comparing by string. This makes an error inssd/utils/nms.py
. So I changed the version comparison code.