问题描述:

I'm working on a Computer Vision system and this is giving me a serious headache. I'm having trouble re-implementing an old gradient operator more efficiently, I'm working with numpy and openCV2.

This is what I had:

`def gradientX(img):`

rows, cols = img.shape

out = np.zeros((rows,cols))

for y in range(rows-1):

Mr = img[y]

Or = out[y]

Or[0] = Mr[1] - Mr[0]

for x in xrange(1, cols - 2):

Or[x] = (Mr[x+1] - Mr[x-1])/2.0

Or[cols-1] = Mr[cols-1] - Mr[cols-2]

return out

def gradient(img):

return [gradientX(img), (gradientX(img.T).T)]

I've tried using numpy's gradient operator but the result is not the same

For this input

`array([[ 3, 4, 5],`

[255, 0, 12],

[ 25, 15, 200]])

Using my `gradient`

returns

`[array([[ 1., 0., 1.],`

[-255., 0., 12.],

[ 0., 0., 0.]]),

array([[ 252., -4., 0.],

[ 0., 0., 0.],

[-230., 15., 0.]])]

While using numpy's `np.gradient`

returns

`[array([[ 252. , -4. , 7. ],`

[ 11. , 5.5, 97.5],

[-230. , 15. , 188. ]]),

array([[ 1. , 1. , 1. ],

[-255. , -121.5, 12. ],

[ -10. , 87.5, 185. ]])]

There are cleary some similarities between the results but they're definitely not the same. So I'm missing something here or the two operators aren't mean to produce the same results. In that case, I wanted to know how to re-implement my gradientX function so it doesn't use that awful looking double loop for traversing the 2-d array using mostly numpy's potency.

I've been working a bit more on this just to find that my mistake. I was skipping last row and last column when iterating. As @wflynny noted, the result was identical except for a row and a column of zeros.

Provided this, the result could not be the same as np.gradient, but with that change, the results are identical, so there's no need to find any other numpy implementation for this.

Answering my own question, a good numpy's implementation for my gradient algorithm would be

```
import numpy as np
def gradientX(img):
return np.gradient(img)[::-1]
```

I'm also posting the working code, just because it shows how numpy's gradient operator works

```
def computeMatXGradient(img):
rows, cols = img.shape
out = np.zeros((rows,cols))
for y in range(rows):
Mr = img[y]
Or = out[y]
Or[0] = float(Mr[1]) - float(Mr[0])
for x in xrange(1, cols - 1):
Or[x] = (float(Mr[x+1]) - float(Mr[x-1]))/2.0
Or[cols-1] = float(Mr[cols-1]) - float(Mr[cols-2])
return out
```

您可能感兴趣的文章：

- loops - Mute Youtube video on Squarespace
- zookeeper single node multiple data is possible?
- java - Passing an array as parameters not working. What could be the error?
- forecasting - R hts group level forecast
- mysql command executescalar returns null in C#
- php - Reading a specific line from a text file
- clr - Any implementation of an Unrolled Linked List in C#?
- Finding Hudson Log Files
- Forward to a payment-gateway together with POST data using cURL (or any other PHP server side solution)
- WCF in Winforms app - is it always single-threaded?

随机阅读：

**推荐内容**-

**热点内容**