From c38fa9d657e6c5b5027c0ce4ead5f05ef76961ec Mon Sep 17 00:00:00 2001 From: zmurez Date: Wed, 19 Aug 2020 23:26:17 -0700 Subject: [PATCH] cv2 resize hangs --- hloc/extract_features.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/hloc/extract_features.py b/hloc/extract_features.py index 39d78ded..e962787b 100644 --- a/hloc/extract_features.py +++ b/hloc/extract_features.py @@ -13,6 +13,8 @@ from .utils.base_model import dynamic_load from .utils.tools import map_tensor +# https://stackoverflow.com/questions/54013846/pytorch-dataloader-stucked-if-using-opencv-resize-method +cv2.setNumThreads(0) ''' A set of standard configurations that can be directly selected from the command