mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-06 14:36:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

dnn: fix dnn python test files

Browse Source
This commit is contained in:
Alexander Alekhin 2017-06-26 15:13:00 +03:00
parent e03569f3a8
commit ee54bafe6b
5 changed files with 14 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/dnn/test/cityscapes_semsegm_test_enet.py
View File

@ -4,13 +4,11 @@ import os
import fnmatch import fnmatch
import argparse import argparse
# sys.path.append('<path to opencv_build_dir/lib>')
sys.path.append('/home/arrybn/build/opencv_w_contrib/lib')
try: try:
import cv2 as cv import cv2 as cv
except ImportError: except ImportError:
raise ImportError('Can\'t find opencv. If you\'ve built it from sources without installation, ' raise ImportError('Can\'t find OpenCV Python module. If you\'ve built it from sources without installation, '
'uncomment the line before and insert there path to opencv_build_dir/lib dir') 'configure environemnt variable PYTHONPATH to "opencv_build_dir/lib" directory (with "python3" subdirectory if required)')
try: try:
import torch import torch
except ImportError: except ImportError:

12
modules/dnn/test/imagenet_cls_test_alexnet.py
View File

@ -5,20 +5,16 @@ import os
import argparse import argparse
import time import time
# sys.path.append('<path to git/caffe/python dir>')
sys.path.append('/home/arrybn/git/caffe/python')
try: try:
import caffe import caffe
except ImportError: except ImportError:
raise ImportError('Can\'t find caffe. If you\'ve built it from sources without installation, ' raise ImportError('Can\'t find Caffe Python module. If you\'ve built it from sources without installation, '
'uncomment the line before and insert there path to git/caffe/python dir') 'configure environemnt variable PYTHONPATH to "git/caffe/python" directory')
# sys.path.append('<path to opencv_build_dir/lib>')
sys.path.append('/home/arrybn/build/opencv_w_contrib/lib')
try: try:
import cv2 as cv import cv2 as cv
except ImportError: except ImportError:
raise ImportError('Can\'t find opencv. If you\'ve built it from sources without installation, ' raise ImportError('Can\'t find OpenCV Python module. If you\'ve built it from sources without installation, '
'uncomment the line before and insert there path to opencv_build_dir/lib dir') 'configure environemnt variable PYTHONPATH to "opencv_build_dir/lib" directory (with "python3" subdirectory if required)')
class DataFetch(object): class DataFetch(object):

12
modules/dnn/test/imagenet_cls_test_googlenet.py
View File

@ -3,20 +3,16 @@ import sys
import os import os
import argparse import argparse
from imagenet_cls_test_alexnet import MeanChannelsFetch, CaffeModel, DnnCaffeModel, ClsAccEvaluation from imagenet_cls_test_alexnet import MeanChannelsFetch, CaffeModel, DnnCaffeModel, ClsAccEvaluation
# sys.path.append('<path to git/caffe/python dir>')
sys.path.append('/home/arrybn/git/caffe/python')
try: try:
import caffe import caffe
except ImportError: except ImportError:
raise ImportError('Can\'t find caffe. If you\'ve built it from sources without installation, ' raise ImportError('Can\'t find Caffe Python module. If you\'ve built it from sources without installation, '
'uncomment the line before and insert there path to git/caffe/python dir') 'configure environemnt variable PYTHONPATH to "git/caffe/python" directory')
# sys.path.append('<path to opencv_build_dir/lib>')
sys.path.append('/home/arrybn/build/opencv_w_contrib/lib')
try: try:
import cv2 as cv import cv2 as cv
except ImportError: except ImportError:
raise ImportError('Can\'t find opencv. If you\'ve built it from sources without installation, ' raise ImportError('Can\'t find OpenCV Python module. If you\'ve built it from sources without installation, '
'uncomment the line before and insert there path to opencv_build_dir/lib dir') 'configure environemnt variable PYTHONPATH to "opencv_build_dir/lib" directory (with "python3" subdirectory if required)')
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()

6
modules/dnn/test/imagenet_cls_test_inception.py
View File

@ -5,13 +5,11 @@ import argparse
import tensorflow as tf import tensorflow as tf
from tensorflow.python.platform import gfile from tensorflow.python.platform import gfile
from imagenet_cls_test_alexnet import MeanValueFetch, DnnCaffeModel, Framework, ClsAccEvaluation from imagenet_cls_test_alexnet import MeanValueFetch, DnnCaffeModel, Framework, ClsAccEvaluation
# sys.path.append('<path to opencv_build_dir/lib>')
sys.path.append('/home/arrybn/build/opencv_w_contrib/lib')
try: try:
import cv2 as cv import cv2 as cv
except ImportError: except ImportError:
raise ImportError('Can\'t find opencv. If you\'ve built it from sources without installation, ' raise ImportError('Can\'t find OpenCV Python module. If you\'ve built it from sources without installation, '
'uncomment the line before and insert there path to opencv_build_dir/lib dir') 'configure environemnt variable PYTHONPATH to "opencv_build_dir/lib" directory (with "python3" subdirectory if required)')
# If you've got an exception "Cannot load libmkl_avx.so or libmkl_def.so" or similar, try to export next variable # If you've got an exception "Cannot load libmkl_avx.so or libmkl_def.so" or similar, try to export next variable
# before runnigng the script: # before runnigng the script:

5
modules/dnn/test/pascal_semsegm_test_fcn.py
View File

@ -5,12 +5,11 @@ import argparse
import time import time
from imagenet_cls_test_alexnet import CaffeModel, DnnCaffeModel from imagenet_cls_test_alexnet import CaffeModel, DnnCaffeModel
sys.path.append('/home/arrybn/build/opencv_w_contrib/lib')
try: try:
import cv2 as cv import cv2 as cv
except ImportError: except ImportError:
raise ImportError('Can\'t find opencv. If you\'ve built it from sources without installation, ' raise ImportError('Can\'t find OpenCV Python module. If you\'ve built it from sources without installation, '
'uncomment the line before and insert there path to opencv_build_dir/lib dir') 'configure environemnt variable PYTHONPATH to "opencv_build_dir/lib" directory (with "python3" subdirectory if required)')
def get_metrics(conf_mat): def get_metrics(conf_mat):