Merge pull request spmallick#26 from vaibhawchandel/master

Faster Selective Search
avarf · Sep 19, 2017 · e0afc12 · e0afc12
2 parents a2454d4 + 29f371a
commit e0afc12
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/SelectiveSearch/ssearch.cpp b/SelectiveSearch/ssearch.cpp
@@ -25,8 +25,16 @@ int main(int argc, char** argv) {
         return -1;
     }
 
+    // speed-up using multithreads
+    setUseOptimized(true);
+    setNumThreads(4);
+
     // read image
     Mat im = imread(argv[1]);
+    // resize image
+    int newHeight = 200;
+    int newWidth = im.cols*newHeight/im.rows;
+    resize(im, im, Size(newWidth, newHeight));
 
     // create Selective Search Segmentation Object using default parameters
     Ptr<SelectiveSearchSegmentation> ss = createSelectiveSearchSegmentation();

diff --git a/SelectiveSearch/ssearch.py b/SelectiveSearch/ssearch.py
@@ -7,7 +7,6 @@
 '''
 
 import sys
-sys.path.insert(0, '/opt/lib/python3.5/dist-packages')
 import cv2
 
 if __name__ == '__main__':
@@ -17,8 +16,16 @@
         print(__doc__)
         sys.exit(1)
 
+    # speed-up using multithreads
+    cv2.setUseOptimized(True);
+    cv2.setNumThreads(4);
+
     # read image
     im = cv2.imread(sys.argv[1])
+    # resize image
+    newHeight = 200
+    newWidth = int(im.shape[1]*200/im.shape[0])
+    im = cv2.resize(im, (newWidth, newHeight))    
 
     # create Selective Search Segmentation Object using default parameters
     ss = cv2.ximgproc.segmentation.createSelectiveSearchSegmentation()