在 Powershell 中使用 CV2 和 Skimage 时出现内存错误

Question

So, I have a python script that uses CV2 and Skimage to compare all images in a filepath and returns the closest match. It is tested and works.

import os
from skimage.metrics import structural_similarity as ssim
import cv2
from shutil import copyfile

def Compare_Images(testDir, reference):

    os.chdir(testDir)
    testlist = os.listdir(testDir)

    smax = 0

    closest = 'null'
    '''s = ssim(imageA, imageB)'''

    for i in range(len(testlist)):

        Reference = cv2.imread(reference)
        Check = cv2.imread(testlist[i])

        Reference = cv2.cvtColor(Reference, cv2.COLOR_BGR2GRAY)
        Check = cv2.cvtColor(Check, cv2.COLOR_BGR2GRAY)

        s= ssim(Reference, Check)
        if (s>smax):
            smax = s
            closest = testlist[i]

    NewPath = "C:\Localdata" + "\\" + closest
    OldPath = testDir + "\\" + closest        
    copyfile(closest, NewPath)
    print (closest)
    return closest

I then modified it to be called from a powershell script (by setting what were input arguments equal to sys.argv[1] for TestDir and sys.argv[2] for reference):

import os
from skimage.metrics import structural_similarity as ssim
import cv2
import sys

from shutil import copyfile



#def Compare_Images(testDir, reference):


testDir = sys.argv[1]


reference = sys.argv[2]



os.chdir(testDir)

testlist = os.listdir(testDir)


smax = 0


closest = 'null'
#s = ssim(imageA, imageB)


for i in range(len(testlist)):
    Reference = cv2.imread(reference)
    Check = cv2.imread(testlist[i])

    Reference = cv2.cvtColor(Reference, cv2.COLOR_BGR2GRAY)
    Check = cv2.cvtColor(Check, cv2.COLOR_BGR2GRAY)

    s= ssim(Reference, Check)
    if (s>smax):
        smax = s
        closest = testlist[i]


print (closest)

copyfile(closest, r"C:\Localdata")

However when I call it from Powershell I get the following error:

    python : Traceback (most recent call last):
At C:\...\pythonpasserTest4.ps1:16 char:1
+ python $arg3 $arg1 $arg2
+ ~~~~~~~~~~~~~~~~~~~~~~~~
    + CategoryInfo          : NotSpecified: (Traceback (most recent call last)::String) [], RemoteException
    + FullyQualifiedErrorId : NativeCommandError

  File "C:\...\ImagecompareV2.py", line 45, in <module>
    s= ssim(Reference, Check)
  File "C:\Program Files (x86)\Python37-32\lib\site-packages\skimage\metrics\_structural_similarity.py", line 205, in structural_similarity
    ux ** 2 + uy ** 2 + C1,
MemoryError: Unable to allocate array with shape (4032, 3024) and data type float64

The PS script is simply:

$env:Path += ";C:\Program Files (x86)\Python37-32";
$env:PATHEXT += ";.py"

$arg1 = 'C:\...\Test Pictures'

$arg2 = 'C:\...\IMG123.JPG'

$arg3 = 'C:\...\ImagecompareV2.py'



python $arg3 $arg1 $arg2 

What can be done to resolve this?

Answer 1

The problem was that I tested it in Spyder, and Powershell was trying to run it in the 32-bit python. Uninstalling 32bit python and installing 64bit python solved the issue.