ETRI IVCL 1.0.0
Acceleration SW Platform for Ondevice
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Functions | Variables
denseUNet_best Namespace Reference

Functions

def make_colorwheel ()
 Public Function ##########. More...
 
def flow_uv_to_colors (u, v, convert_to_bgr=False)
 
def flow_to_color (flow_uv, clip_flow=None, convert_to_bgr=False)
 
def trans_SAIs_to_LF (imgSAIs)
 
def trans_order (imgSAIs)
 
def shift_value_5x5 (i, shift_val)
 
def index_picker_5x5 (i)
 
def image_data_center (batch_size=1, data_path=None, source='train_source')
 
def image_data (batch_size=1, data_path=None, source='train_source', sai=25)
 
def input_shifting (center=None, batch_size=1, sai=27, shift_val=0.7)
 
def slice_warp (shift, flow_h, flow_v, sai)
 
def slice_warp2 (shift, flow_h, flow_v, sai)
 
def change_order (imgs, batch_size)
 
def l_loss (imgs, lables, batch_size)
 
def denseUNet (input, batch_size)
 Fundamental Network ##########. More...
 
def denseUNet_train (script_path=None, data_path=None, tot=1, sai=25, shift_val=0.7, batch_size=1)
 Network Generator ##########. More...
 
def denseUNet_test (script_path=None, data_path=None, tot=1, sai=25, shift_val=0.7, batch_size=1)
 
def denseUNet_deploy (script_path=None, batch_size=1, sai=25, shift_val=0.7)
 
def denseUNet_solver (script_train_path=None, script_test_path=None, solver_path=None, snapshot_path=None)
 Solver Generator ##########. More...
 
def denseUNet_runner (script_path=None, model_path=None, src_color=None, n=None)
 Runner ##########. More...
 
def denseUNet_tester (script_path=None, model_path=None, dataset_path=None, output_predict_path=None, output_GT_path=None, test_range=1)
 Tester ##########. More...
 

Variables

 threshold
 
string TRAINSET_PATH = './datas/face_dataset/face_train_9x9'
 Main ##########. More...
 
string TESTSET_PATH = './datas/face_dataset/face_test_9x9'
 
string MODEL_PATH = './backup/denseUNet_best.caffemodel'
 
string TRAIN_PATH = './scripts/denseUNet_train.prototxt'
 
string TEST_PATH = './scripts/denseUNet_test.prototxt'
 
string DEPLOY_PATH = './scripts/denseUNet_deploy.prototxt'
 
string SOLVER_PATH = './scripts/denseUNet_solver.prototxt'
 
string OUTPUT_PREDICT = './output/predict'
 
string OUTPUT_GT = './output/GT'
 
int TRAIN_TOT = 1207
 
int SAI = 25
 
float SHIFT_VAL = 0.77625
 
string MOD = 'test'
 
 script_path
 
 data_path
 
 tot
 
 sai
 
 batch_size
 
 shift_val
 
 script_train_path
 
 script_test_path
 
 solver_path
 
 snapshot_path
 
 solver = caffe.get_solver(SOLVER_PATH)
 
 model_path
 
 dataset_path
 
 output_predict_path
 
 output_GT_path
 
 test_range
 
 test_img = cv2.imread('./test.png', 1)
 
 test_list
 
 flow_color_list
 
 src_color
 
 n
 

Function Documentation

◆ change_order()

def denseUNet_best.change_order (   imgs,
  batch_size 
)

◆ denseUNet()

def denseUNet_best.denseUNet (   input,
  batch_size 
)

Fundamental Network ##########.

◆ denseUNet_deploy()

def denseUNet_best.denseUNet_deploy (   script_path = None,
  batch_size = 1,
  sai = 25,
  shift_val = 0.7 
)

◆ denseUNet_runner()

def denseUNet_best.denseUNet_runner (   script_path = None,
  model_path = None,
  src_color = None,
  n = None 
)

Runner ##########.

◆ denseUNet_solver()

def denseUNet_best.denseUNet_solver (   script_train_path = None,
  script_test_path = None,
  solver_path = None,
  snapshot_path = None 
)

Solver Generator ##########.

◆ denseUNet_test()

def denseUNet_best.denseUNet_test (   script_path = None,
  data_path = None,
  tot = 1,
  sai = 25,
  shift_val = 0.7,
  batch_size = 1 
)

◆ denseUNet_tester()

def denseUNet_best.denseUNet_tester (   script_path = None,
  model_path = None,
  dataset_path = None,
  output_predict_path = None,
  output_GT_path = None,
  test_range = 1 
)

Tester ##########.

◆ denseUNet_train()

def denseUNet_best.denseUNet_train (   script_path = None,
  data_path = None,
  tot = 1,
  sai = 25,
  shift_val = 0.7,
  batch_size = 1 
)

Network Generator ##########.

◆ flow_to_color()

def denseUNet_best.flow_to_color (   flow_uv,
  clip_flow = None,
  convert_to_bgr = False 
)

◆ flow_uv_to_colors()

def denseUNet_best.flow_uv_to_colors (   u,
  v,
  convert_to_bgr = False 
)

◆ image_data()

def denseUNet_best.image_data (   batch_size = 1,
  data_path = None,
  source = 'train_source',
  sai = 25 
)

◆ image_data_center()

def denseUNet_best.image_data_center (   batch_size = 1,
  data_path = None,
  source = 'train_source' 
)

◆ index_picker_5x5()

def denseUNet_best.index_picker_5x5 (   i)

◆ input_shifting()

def denseUNet_best.input_shifting (   center = None,
  batch_size = 1,
  sai = 27,
  shift_val = 0.7 
)

◆ l_loss()

def denseUNet_best.l_loss (   imgs,
  lables,
  batch_size 
)

◆ make_colorwheel()

def denseUNet_best.make_colorwheel ( )

Public Function ##########.

◆ shift_value_5x5()

def denseUNet_best.shift_value_5x5 (   i,
  shift_val 
)

◆ slice_warp()

def denseUNet_best.slice_warp (   shift,
  flow_h,
  flow_v,
  sai 
)

◆ slice_warp2()

def denseUNet_best.slice_warp2 (   shift,
  flow_h,
  flow_v,
  sai 
)

◆ trans_order()

def denseUNet_best.trans_order (   imgSAIs)

◆ trans_SAIs_to_LF()

def denseUNet_best.trans_SAIs_to_LF (   imgSAIs)

Variable Documentation

◆ batch_size

denseUNet_best.batch_size

◆ data_path

denseUNet_best.data_path

◆ dataset_path

denseUNet_best.dataset_path

◆ DEPLOY_PATH

denseUNet_best.DEPLOY_PATH = './scripts/denseUNet_deploy.prototxt'

◆ flow_color_list

denseUNet_best.flow_color_list

◆ MOD

string denseUNet_best.MOD = 'test'

◆ MODEL_PATH

denseUNet_best.MODEL_PATH = './backup/denseUNet_best.caffemodel'

◆ model_path

denseUNet_best.model_path

◆ n

denseUNet_best.n

◆ OUTPUT_GT

string denseUNet_best.OUTPUT_GT = './output/GT'

◆ output_GT_path

denseUNet_best.output_GT_path

◆ OUTPUT_PREDICT

string denseUNet_best.OUTPUT_PREDICT = './output/predict'

◆ output_predict_path

denseUNet_best.output_predict_path

◆ SAI

denseUNet_best.SAI = 25

◆ sai

denseUNet_best.sai

◆ script_path

denseUNet_best.script_path

◆ script_test_path

denseUNet_best.script_test_path

◆ script_train_path

denseUNet_best.script_train_path

◆ SHIFT_VAL

float denseUNet_best.SHIFT_VAL = 0.77625

◆ shift_val

denseUNet_best.shift_val

◆ snapshot_path

denseUNet_best.snapshot_path

◆ solver

denseUNet_best.solver = caffe.get_solver(SOLVER_PATH)

◆ SOLVER_PATH

denseUNet_best.SOLVER_PATH = './scripts/denseUNet_solver.prototxt'

◆ solver_path

denseUNet_best.solver_path

◆ src_color

denseUNet_best.src_color

◆ test_img

denseUNet_best.test_img = cv2.imread('./test.png', 1)

◆ test_list

denseUNet_best.test_list

◆ TEST_PATH

denseUNet_best.TEST_PATH = './scripts/denseUNet_test.prototxt'

◆ test_range

denseUNet_best.test_range

◆ TESTSET_PATH

denseUNet_best.TESTSET_PATH = './datas/face_dataset/face_test_9x9'

◆ threshold

denseUNet_best.threshold

◆ tot

denseUNet_best.tot

◆ TRAIN_PATH

denseUNet_best.TRAIN_PATH = './scripts/denseUNet_train.prototxt'

◆ TRAIN_TOT

denseUNet_best.TRAIN_TOT = 1207

◆ TRAINSET_PATH

denseUNet_best.TRAINSET_PATH = './datas/face_dataset/face_train_9x9'

Main ##########.