• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• DL.EyeSight-master
  • Others
    • vedia
      • show.py
      • __init__.py
      • convert2voc.py
    • satellite
      • prepare_trainsamples.py
      • __init__.py
      • clip_video.py
      • bbox_cluster.py
      • process.py
    • lsd12
      • format_input.py
      • check_dataset.py
      • label_config.py
      • __init__.py
    • __init__.py
    • README.md
    • voc
      • process_pascal_voc.py
      • __init__.py
  • results
    • test_res_image
    • README.md
  • examples
    • yolo
      • train.py
      • __init__.py
    • __init__.py
    • ssd
      • squeezenet_300_trainer.py
      • squeezenet_512_trainer.py
      • __init__.py
      • vgg_trainer.py
    • unet
      • predict.py
      • train.py
      • __init__.py
  • LICENSE
  • eagle
    • brain
      • solver
        • yolo_u_solver.py
        • solver.py
        • ssd_solver.py
        • yolo_solver.py
        • __init__.py
      • yolo
        • yolo_u_net.py
        • yolo_tiny_net.py
        • net.py
        • __init__.py
        • yolo_net.py
      • rotation
        • yolo
          • net.py
          • __init__.py
          • rotation_tiny_net.py
        • __init__.py
      • __init__.py
      • ssd
        • box_encode_decode_utils.py
        • models
          • vgg.py
          • squeezenet_512.py
          • vgg_dilated.py
          • components.py
          • net.py
          • __init__.py
          • squeezenet_300.py
        • normalization.py
        • loss.py
        • __init__.py
        • anchor_boxes.py
    • observe
      • base
        • basebatch.py
        • meta.py
        • __init__.py
        • basetype.py
      • augmentors
        • color.py
        • arithmetic.py
        • __init__.py
        • flip.py
        • blur.py
      • __init__.py
    • __init__.py
    • README.md
    • utils.py
    • parameter.py
  • datum
    • utils
      • process_config.py
      • tools.py
      • __init__.py
    • models
      • yolo
        • yolo_dataset.py
        • yolo_batch_dataset.py
        • __init__.py
      • __init__.py
      • ssd
        • ssd_dataset.py
        • box_encoder.py
        • __init__.py
    • __init__.py
    • README.md
    • meta
      • __init__.py
      • dataset.py
  • README.md
  • conf
    • ssd_train_512.cfg
    • yolo_train_server.cfg
    • ssd_train_server.cfg
    • ssd_train.cfg
    • yolo_unet_train_server.cfg
    • yolo_unet_train.cfg
    • dilated_ssd_train.cfg
    • yolo_train.cfg
  • checks
    • datasets
      • __init__.py
      • check_ssd_dataset.py
    • brain
      • yolo
        • check_yolo_model.py
        • __init__.py
      • __init__.py
      • ssd
        • __init__.py
        • check_ssd_model.py
    • observe
      • check_average_blur.py
      • check_gaussian_blur.py
      • check_color.py
      • check_background.py
      • check_median_blur.py
      • __init__.py
      • check_parameters.py
    • __init__.py

# Copyright (c) 2009 IW.
# All rights reserved.
#
# Author: liuguiyang <liuguiyangnwpu@gmail.com>
# Date:   2018/2/28

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import cv2
import numpy as np
from skimage import data

import eagle.utils as eu
from eagle.observe.augmentors.flip import Fliplr
from eagle.observe.augmentors.arithmetic import Add
from eagle.observe.augmentors.color import WithChannels, WithColorspace

TIME_PER_STEP = 10000


def main_WithChannels():
    image = data.astronaut()
    print("image shape:", image.shape)
    print("Press any key or wait %d ms to proceed to the next image." % (TIME_PER_STEP,))

    children_all = [
        ("hflip", Fliplr(1)),
        ("add", Add(50))
    ]

    channels_all = [
        None,
        0,
        [],
        [0],
        [0, 1],
        [1, 2],
        [0, 1, 2]
    ]

    cv2.namedWindow("aug", cv2.WINDOW_NORMAL)
    cv2.imshow("aug", image[..., ::-1])
    cv2.waitKey(TIME_PER_STEP)

    for children_title, children in children_all:
        for channels in channels_all:
            aug = WithChannels(channels=channels, children=children)
            img_aug = aug.augment_image(image)
            print("dtype", img_aug.dtype, "averages", np.average(img_aug, axis=tuple(range(0, img_aug.ndim-1))))
            #print("dtype", img_aug.dtype, "averages", img_aug.mean(axis=range(1, img_aug.ndim)))

            # title = "children=%s | channels=%s" % (children_title, channels)
            # img_aug = ia.draw_text(img_aug, x=5, y=5, text=title)

            cv2.imshow("aug", img_aug[..., ::-1]) # here with rgb2bgr
            cv2.waitKey(TIME_PER_STEP)


def main_WithColorspace():
    image = data.astronaut()
    print("image shape:", image.shape)

    aug = WithColorspace(
        from_colorspace="RGB",
        to_colorspace="HSV",
        children=WithChannels(0, Add(50))
    )

    aug_no_colorspace = WithChannels(0, Add(50))

    img_show = np.hstack([
        image,
        aug.augment_image(image),
        aug_no_colorspace.augment_image(image)
    ])

    cv2.namedWindow("aug", cv2.WINDOW_NORMAL)
    cv2.imshow("aug", img_show[..., ::-1])
    cv2.waitKey(TIME_PER_STEP)

if __name__ == "__main__":
    # main_WithChannels()
    main_WithColorspace()