gem-cmarkers.py 566 B

1234567891011121314151617181920212223242526272829303132
  1. #!/usr/bin/env python
  2. import sys
  3. from cv2 import imwrite
  4. from numpy import mean, binary_repr, ones
  5. from numpy.random import randint
  6. from scipy.ndimage import zoom
  7. for i in range(0, 16):
  8. img = ones((6, 6))*255
  9. img[1, 1] = 0
  10. img[4, 1] = 0
  11. img[1, 4] = 0
  12. if i%2 == 1:
  13. img[2, 2] = 0
  14. if (i>>1)%2 == 1:
  15. img[2, 3] = 0
  16. if (i>>2)%2 == 1:
  17. img[3, 2] = 0
  18. if (i>>3)%2 == 1:
  19. img[3, 3] = 0
  20. print (img)
  21. marker = zoom(img, zoom=50, order=0)
  22. imwrite('marker_images/marker_{}.png'.format(i), marker)