FilesExpand file tree

 master

/

generate_input.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

19 lines (15 loc) · 402 Bytes

 master

/

generate_input.py

Copy path

Top

File metadata and controls

• Code
• Blame

19 lines (15 loc) · 402 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

import cv2

n = 20

im = cv2.imread('input/ship.jpg')

h, w, a = im.shape

h_unit = h // n; w_unit = w // n

for i in range(n):

start_x = h_unit * i

for j in range(n):

#print(i, j)

start_y = w_unit * j

cv2.imwrite(

'input/large_input/ship_%i_%i.jpg' % (i, j),

im[start_x : start_x + h_unit, start_y : start_y + w_unit, :]

)

print('Done.')