def parse(dataset, location, wRatio, hRatio):
print('file is {} and headers are {}'.format(location, next(dataset)))
next(dataset)
vals = list(enumerate(dataset))
for idx, row in vals:
print('{}/{}'.format(idx, len(vals)), sep=' ', end='\r', flush=True)
url, name, x1, x2, y1, y2 = row
try:
image = iu.loadImage(url)
image = smartRectSelector(
image, getRealValue(x1, len(image[0])),
getRealValue(x2, len(image[0]), len(image[0])),
getRealValue(y1, len(image)),
getRealValue(y2, len(image), len(image)), wRatio, hRatio)
cv2.imwrite(location + '/{}.jpg'.format(idx),
cv2.resize(image, (400, 400)))
except:
print('error: ', idx, ' row: ', row)
def _binaryzation(self):
self.img = ImageUtil.binaryzation(self.img)
self._refreshBoard()
def _invert(self):
self.img = ImageUtil.invert(self.img)
self._refreshBoard()
def _emboss(self):
self.img = ImageUtil.emboss(self.img)
self._refreshBoard()
def _sharpen(self):
self.img = ImageUtil.sharpen(self.img)
self._refreshBoard()
def _gray(self):
self.img = ImageUtil.gray(self.img)
self._refreshBoard()
def _canny(self):
self.img = ImageUtil.canny(self.img)
self._refreshBoard()
def _blur(self):
self.img = ImageUtil.blur(self.img)
self._refreshBoard()
def _drawBucket(self, event):
boardPos = self._getPosFromGlobal(event.pos())
fillPositions = ImageUtil.floodFill(self.img, boardPos)
painter = self._initPainter()
[painter.drawPoint(x, y) for x, y in fillPositions]
def _adjustSaturation(self, value):
self.adjusting = True
self.bufferImg = ImageUtil.adjustSaturation(self.img, value)
self.update()
def _adjustWarm(self, value):
self.adjusting = True
self.bufferImg = ImageUtil.adjustWarm(self.img, value)
self.update()