Live In Your Style

This is project Live In Your Style, which has objective

• Segment person in an image frame and style transfer background
• Integrate with cloud computing and edge application

This repository is the source code for training required deep learning model. And its integration applications.

Code base structure

In this code base, it maintains the code for two model-related tasks

• Segmentation (2 classes, background and person)
• Style transfer

In the top folder, it has subdirectories

• Libs

  The required libraries or model code will be in here. For example, because we use DeeplabV3+ model and Slim library from Tensorflow, a copy of their code base will be placed in here.

• Segmentation

  This could be a workspace for the preparation of segmentation model. It may contains the training, validation, test tools.

• StyleTransfer

  Similar to Segmentation folder, this folder is a workspace for the preparation of style transfer model.

Workspace Environment

For each workspace folder, Segmentation and StyleTransfer, it has 2 shell scripts, under bin folder, to activate and deactivate the working environment.

• activate.sh
• deactivate.sh

After launching the activate.sh script

$ source activate.sh

current environment will be activated to facilitate working on model-related tasks by

• Set proper environment variables

  • TOP

    It stores the full path of this code base. You could inspect it by

    $ echo $TOP 
    /home/jovyan/AT073_10_Orig_Style

    Or use it to refer your file in your code base, for example

    $ python ${TOP}/Segmentation/bin/train.py

  • WORKENV

    It stores string about the variant of activated working environment, it may be one of Segmentation or StyleTransfer. You could inspect it by

    $ echo $WORKENV
    Segmentation

• Add the path of required Python libraries in PYTHONPATH
• Create shortcuts, in bin folder, for useful tools

To reset the activated working environment, just run as below

$ source ${TOP}/${WORKENV}/bin/deactivate.sh

It will clear stuff that the activate.sh setup.