diff options
| -rw-r--r-- | pymba/vimbaframe.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/pymba/vimbaframe.py b/pymba/vimbaframe.py index 5735430..fdd0d49 100644 --- a/pymba/vimbaframe.py +++ b/pymba/vimbaframe.py @@ -4,7 +4,11 @@ from vimbaexception import VimbaException from vimbadll import VimbaDLL from vimbadll import VimbaC_MemoryBlock from ctypes import * -import numpy as np +try: + import numpy as np +except ImportError: + print('warning: numpy not found, some VimbaFrame methods will not be available') + """ Map pixel formats to bytes per pixel. @@ -162,5 +166,8 @@ class VimbaFrame(object): def getImage(self): cframe = self._frame data = cast(cframe.buffer, POINTER(c_ubyte * cframe.imageSize)) - return np.ndarray(buffer=data.contents, dtype=np.uint8, shape=(cframe.height, cframe.width)) - + try: + return np.ndarray(buffer=data.contents, dtype=np.uint8, shape=(cframe.height, cframe.width)) + except NameError as e: + print('install numpy to use this method or use getBufferByteData instead') + raise e |