Don't load the entire torchvision library just for nms when the function is already in the torch library.
rafale77