FilesExpand file tree

 master

/

makeDataSetOCR.js

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

44 lines (39 loc) · 1.56 KB

 master

/

makeDataSetOCR.js

Copy path

Top

File metadata and controls

• Code
• Blame

44 lines (39 loc) · 1.56 KB

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

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

const fs = require('fs');

const cv = require('../');

const labeledDataPath = '../data/ocr-nocommit/letters';

const outputDataPath = '../data/ocr-nocommit/letters_generated';

const lccs = Array(26).fill(97).map((v, i) => v + i).map(a => String.fromCharCode(a));

const blur = img => img.blur(new cv.Size(8, 8), 1, 1);

const invert = img => img.threshold(254, 255, cv.THRESH_BINARY_INV);

const generate = (img, clazz, nr) => {

for (let angle = 0; angle <= 60; angle += 10) {

const rotAngle = -30 + angle;

const rotMat = cv.getRotationMatrix2D(new cv.Point(img.cols / 2, img.rows / 2), rotAngle);

const rotated = invert(img).warpAffine(rotMat);

for (let weight = 0; weight <= 3; weight += 1) {

const threshWeight = 200 - (weight * 50);

const result = blur(rotated)

.threshold(threshWeight, 255, cv.THRESH_BINARY_INV);

cv.imwrite(`${outputDataPath}/${clazz}/${clazz}_${nr}_w${weight}_r${angle}.png`, result.resize(40, 40));

}

}

};

/*

lccs.forEach((clazz) => {

for (let nr = 0; nr < 10; nr += 1) {

const img = cv.imread(`${labeledDataPath}/${clazz}/${clazz}${nr}.png`);

generate(img, clazz, nr);

}

});

*/

const makeGrid = (clazz) => {

const dir = `${outputDataPath}/${clazz}`;

const gridMat = new cv.Mat(10 * 40, 28 * 40, cv.cvTypes.CV_8UC3);

const files = fs.readdirSync(dir);

files.forEach((file, i) => {

const x = (i % 28) * 40;

const y = parseInt(i / 28) * 40;

cv.imread(`${dir}/${file}`).copyTo(gridMat.getRegion(new cv.Rect(x, y, 40, 40)));

});

cv.imwrite(`${outputDataPath}/${clazz}_grid.png`, gridMat);

};