core.occusion_handing.img_combination

Classes

ImgCombination

Functions

combine_bg_with_obj_single_refine 
combine_bg_with_obj_single_refine(center: numpy.ndarray, obj: numpy.ndarray, bg: numpy.ndarray, mask: numpy.ndarray) -> numpy.ndarray
Parameters:
  • center (ndarray) –
  • obj (ndarray) –
  • bg (ndarray) –
  • mask (ndarray) –
Returns:
  • ndarray –
img_combination_with_objs 
img_combination_with_objs(img_bg: numpy.ndarray, img_objs: List[numpy.ndarray], obj_centers_in_img: List[List[int]], index: numpy.ndarray, refine: numpy.ndarray = True) -> numpy.ndarray

将背景图片和物体图片组合

Parameters:
  • img_bg (ndarray) –

    背景图片

  • img_objs (List[ndarray]) –

    物体图片

  • obj_centers_in_img (List[List[int]]) –

    物体图片在背景图片中的中心点

  • index (ndarray) –

    根据mesh到源点距离，由大到小排序的索引

  • refine –

    是否改进
Returns:
  • ndarray –

    组合图片，组合并改进的图片
img_combination_with_one_obj 
img_combination_with_one_obj(center: numpy.ndarray, obj: numpy.ndarray, bg: numpy.ndarray) -> numpy.ndarray

将一个物体图片插入到背景图片中

Parameters:
  • center (ndarray) –

    物体在图片的中心

  • obj (ndarray) –

    物体图片

  • bg (ndarray) –

    背景图片
Returns:
  • ndarray –

    组合图片