Advertisement
Guest User

Untitled

a guest
Feb 23rd, 2020
90
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.01 KB | None | 0 0
  1. def getSpacing(im):
  2. l = im.shape[0]
  3. vote = np.zeros((l, l // 5 + 1))
  4. th = im.shape[1]
  5. result = list()
  6.  
  7. for i in range(im.shape[0]):
  8. for j in range(im.shape[1]):
  9. if im[i][j] == 1:
  10. for spacing in range(1, l // 5 + 1):
  11. for mul in range(5):
  12. startLine = i - mul * spacing
  13. if startLine >= 0:
  14. vote[startLine][spacing] += 1
  15.  
  16. startLine = -1
  17. tmpV = 0
  18. for i in range(vote.shape[0]):
  19. if np.sum(vote[i]) > tmpV:
  20. startLine = i
  21. tmpV = np.sum(vote[i])
  22. # print("StartLine = ", startLine)
  23. for i in range(vote.shape[0]):
  24. if np.sum(vote[1]) > tmpV + th:
  25. result.append(i)
  26. spacing = -1
  27. tmpV = 0
  28. for j in result:
  29. for i in range(vote.shape[1]):
  30. if vote[j][i] > tmpV:
  31. spacing = i
  32. tmpV = vote[j][i]
  33. result.append(spacing)
  34. return vote
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement