libcom 0.1.1.post2__tar.gz → 0.1.2__tar.gz

{libcom-0.1.1.post2 → libcom-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libcom
-Version: 0.1.1.post2
+Version: 0.1.2
 Summary: Image Composition Toolbox
 Home-page: https://github.com/bcmi/libcom
 Author: BCMI Lab

{libcom-0.1.1.post2 → libcom-0.1.2}/libcom.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libcom
-Version: 0.1.1.post2
+Version: 0.1.2
 Summary: Image Composition Toolbox
 Home-page: https://github.com/bcmi/libcom
 Author: BCMI Lab

{libcom-0.1.1.post2 → libcom-0.1.2}/libcom.egg-info/SOURCES.txt

@@ -132,6 +132,7 @@ libcom/utils/model_download.py
 libcom/utils/process_image.py
 tests/test_color_transfer.py
 tests/test_composite_generation.py
+tests/test_controllable_composition.py
 tests/test_flux_kontext.py
 tests/test_fopa_heat_map.py
 tests/test_fos_score.py

{libcom-0.1.1.post2 → libcom-0.1.2}/setup.py

@@ -109,7 +109,7 @@ if __name__ == '__main__':
     cur_dir = os.path.dirname(os.path.abspath(__file__))
     setup(
         name='libcom',
-        version='0.1.1.post2',
+        version='0.1.2',
         description='Image Composition Toolbox',
         long_description=readme(),
         long_description_content_type='text/markdown',

libcom-0.1.2/tests/test_controllable_composition.py

@@ -0,0 +1,39 @@
+from common import *
+from libcom import ControlComModel
+import os
+import cv2
+import shutil
+import torch
+
+task_name = 'controllable_composition'
+
+def inference_on_single_pair(net, pair, task):
+    bg_img, bbox = pair['background'], pair['bbox']
+    fg_img, fg_mask = pair['foreground'], pair['foreground_mask']
+    comp_img  = net(bg_img, fg_img, bbox, fg_mask, task=task, sample_steps=50, seed=32)
+    img_name  = os.path.basename(bg_img).replace('.png', '.jpg')
+    bbox_bg   = draw_bbox_on_image(bg_img, bbox)
+    grid_img  = make_image_grid([bbox_bg, fg_img, comp_img[0], comp_img[1]])
+    res_path  = os.path.join(result_dir, img_name)
+    cv2.imwrite(res_path, grid_img)
+    print('save result to ', res_path)
+
+if __name__ == '__main__':
+    # collect pairwise test samples
+    test_set = get_controlcom_test_list()
+    result_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'results', task_name)
+    if os.path.exists(result_dir):
+        shutil.rmtree(result_dir)
+    os.makedirs(result_dir, exist_ok=True)
+    print(f'begin testing {task_name}...')
+    # build your model
+    net = ControlComModel(device=0, model_type='ControlCom')
+    inference_on_single_pair(net, test_set[2], ['blending', 'harmonization'])
+    del net
+    net = ControlComModel(device=0, model_type='ControlCom_blend_harm')
+    inference_on_single_pair(net, test_set[1], ['blending', 'harmonization'])
+    del net
+    net = ControlComModel(device=0, model_type='ControlCom_view_comp')
+    inference_on_single_pair(net, test_set[0], ['viewsynthesis', 'composition'])
+    del net
+    print(f'end testing {task_name}!\n')