def __init__(self, file, dstImgPath): self.image = loadImg(file) self.fileName = os.path.basename(file) self.imgName = self.fileName[:self.fileName.rfind('.')] print('ImageColorAug image:', file, self.fileName, self.imgName) assert (self.image is not None) self.dstImgPath = self._createPath(dstImgPath)
def generateImageLabel(imgPath, annotPath, dst): deleteFolder(dst) createPath(dst) for i in listFile(imgPath, 'png'): H, W = getImgHW(loadImg(i)) # print(H,W) fAnnot = getImgAnnotFile(annotPath, i) fAnnotName = getFileName(fAnnot) dstFile = dst + '\\' + fAnnotName print('fAnnot=', fAnnot) # print('dstFile=', dstFile) deleteFile(dstFile) coordinates = getFileCoordinates(fAnnot) writeToAnnotFile(H, W, dstFile, coordinates)
def demonstratePrediction(file, gtMaskFile=None): img = loadImg(file) # img = loadGrayImg(file) simg = blurImg(img) pMask = getPredictImg(simg) pImg = maskToOrignimalImg(img, pMask) if gtMaskFile is not None: maskImg = loadImg(gtMaskFile) # maskImg = resizeImg(maskImg,256,256) maskImg = processMaskImg(maskImg) # mImg = maskToOrignimalImg(img, maskImg) ls, nameList = [], [] ls.append(img), nameList.append('Original') if gtMaskFile is not None: ls.append(maskImg), nameList.append('GT mask') # ls.append(mImg),nameList.append('GT Segmentation') ls.append(pImg), nameList.append('Predict Segmentation') if gtMaskFile is not None: ls.append(pMask), nameList.append('Predict Mask') plotImagList(ls, nameList, gray=True, showticks=False)
def main(): args = cmd_line() src = args.src dst = args.dst createPath(dst) print('src=', src, 'dst=', dst) for i in pathsFiles(src, 'jpg'): # png fileName = getFileName(i) img = loadImg(i) dstFile = args.dst + '\\' + fileName print(i, fileName, dstFile) predImg = getPredictionMaskImg(img) writeImg(predImg, dstFile)
def testFileLabel(imgPath, LabelPath, dstRecImgPath): createPath(dstRecImgPath) for i in listFile(imgPath): imgFile = getFileName(i) img = loadImg(i) H, W = getImgHW(img) print(i, imgFile, H, W) labelFile = LabelPath + '\\' + imgFile[:imgFile.rfind('.')] + '.txt' # labels = getLabelFileLabels(labelFile) # print(labels) coordinats = getCoordinatesFromLabels(H, W, labelFile) # print(coordinats) recImg = rectangleImgFromCoordinates(img.copy(), coordinats) destFile = dstRecImgPath + '\\' + imgFile[:imgFile.rfind('.')] + '_rec' + '.png' print(destFile) writeImg(recImg, destFile)
def getImageMask(file, maskeFile): img = loadImg(file) mask = loadImg(maskeFile) mask = processMaskImg(mask, backColor=0) # added_image = cv2.addWeighted(background, 1.0, mask, 0.8, 0) return img, mask