The TorchVision's version check uses dictionary order. When the version number is '0.17.0+cpu', an incorrect judgment occurs, causing the code to enter the else branch, throwing an error and interrupting the program abnormally.
incorrect code in ssd/utils/nms.py:
if torchvision.__version__ >= '0.3.0':
_nms = torchvision.ops.nms
else:
warnings.warn('No NMS is available. Please upgrade torchvision to 0.3.0+')
sys.exit(-1)
my suggestion is import package : packaging, and than:
from packaging import version
if version.parse(torchvision.__version__) >= version.parse('0.3.0'):
_nms = torchvision.ops.nms
else:
warnings.warn('No NMS is available. Please upgrade torchvision to 0.3.0+')
sys.exit(-1)
The TorchVision's version check uses dictionary order. When the version number is '0.17.0+cpu', an incorrect judgment occurs, causing the code to enter the else branch, throwing an error and interrupting the program abnormally. incorrect code in ssd/utils/nms.py:
my suggestion is import package : packaging, and than: