test_imgAug.py

'''
@lanhuage: python
@Descripttion: image augmentation with labels.
@version: beta
@Author: xiaoshuyui
@Date: 2020-07-17 15:49:30
LastEditors: xiaoshuyui
LastEditTime: 2022-07-02 19:59:08
'''
import sys

sys.path.append("..")
import os

from convertmask.utils.methods.get_multi_shapes import getMultiShapes
from convertmask.utils.auglib.img_aug import (aug_labelimg, aug_labelme, imgFlip,
                                      imgNoise, imgRotation, imgTranslation,
                                      imgZoom)
from skimage import io

BASE_DIR = os.path.abspath(os.path.dirname(os.getcwd())) + os.sep + 'static'
# print(BASE_DIR)
imgPath = BASE_DIR + os.sep + 'test702.jpg'
labelPath = BASE_DIR + os.sep + 'test702.json'
yamlPath = BASE_DIR + os.sep + 'test702.yaml'

imgPath2 = BASE_DIR + os.sep + 'label_255.png'
labelPath2 = BASE_DIR + os.sep + 'label_255.xml'

if __name__ == "__main__":

    #### test1

    imgFlip(imgPath, labelPath)

    imgNoise(imgPath, labelPath)

    imgRotation(imgPath, labelPath)

    imgTranslation(imgPath, labelPath)

    imgZoom(imgPath, labelPath, 1.2)

    #### test2

    n = imgNoise(imgPath, labelPath, flag=False)

    tmp = n['noise']

    img, processedImg = tmp.oriImg, tmp.processedImg

    r = imgRotation(img, processedImg, flag=False, angle=15)

    tmp = r['rotation']

    img, processedImg = tmp.oriImg, tmp.processedImg

    t = imgTranslation(img, processedImg, flag=False)

    tmp = t['trans']

    img, processedImg = tmp.oriImg, tmp.processedImg

    f = imgFlip(img, processedImg, flag=False)

    tmp = f['h_v']

    img, processedImg = tmp.oriImg, tmp.processedImg

    parent_path = os.path.dirname(imgPath)

    if os.path.exists(parent_path + os.sep + 'jsons_'):
        pass
    else:
        os.makedirs(parent_path + os.sep + 'jsons_')
    fileName = 'test'
    io.imsave(
        parent_path + os.sep + 'jsons_' + os.sep + fileName + '_assumble.jpg',
        img)

    assumbleJson = getMultiShapes(parent_path + os.sep + 'jsons_' + os.sep +
                                  fileName + '_assumble.jpg',
                                  processedImg,
                                  flag=True,
                                  labelYamlPath='')

    saveJsonPath = parent_path + os.sep + 'jsons_' + os.sep + fileName + '_assumble.json'

    with open(saveJsonPath, 'w') as f:
        f.write(assumbleJson)

    #### test3

    aug_labelme(imgPath, labelPath,yamlFilePath=yamlPath)

    #### test4

    # aug_labelimg(imgPath2, labelPath2)