Explorar o código

服装版拍摄

rambo hai 1 mes
pai
achega
5a6ab4d776

+ 2 - 1
python/api.py

@@ -768,7 +768,7 @@ async def _process_model_images(aigc_clazz, run_main, return_data_check_before_d
                     continue
                     
                 shutil.copy(ceshi_image_path, save_image_path)
-                image_path = aigc_clazz.generateUpperShoes(save_image_path, model_id, save_image_path)
+                image_path = aigc_clazz.generateModelFitting(save_image_path, model_id, face_type,save_image_path)
                 
                 goods_art_dict_info_item["模特图"] = image_path
                 new_goods_dict[goods_art_no_info]["货号资料"][idx_key] = goods_art_dict_info_item
@@ -1383,6 +1383,7 @@ def sync_sys_configs(params: SyncLocalConfigs):
     # 追加配置参数 machine_type 拍照机设备类型;0鞋;1服装
     url = settings.DOMAIN + f"/api/ai_image/camera_machine/get_all_user_configs?machine_type={MACHINE_TYPE}"
     result = requests.get(url=url, headers=headers)
+    print("result",result.json())
     sys_configs = result.json().get("data", {}).get("configs")
     session = SqlQuery()
     sysConfigs = CRUD(SysConfigs)

+ 1 - 1
python/config.ini

@@ -10,7 +10,7 @@ app_run=api:app
 # 端口号
 port=7074
 debug=false
-env=prod
+env=dev
 # 线程数
 works=1
 project=惠利玛

+ 1 - 1
python/databases.py

@@ -181,7 +181,7 @@ class CRUD:
                 )
             )
         )
-        print("SQL 打印==>",str(query))
+        # print("SQL 打印==>",str(query))
         result = session.exec(query).first()
         if result:
             for key, value in kwargs.items():

+ 6 - 4
python/service/base_deal.py

@@ -349,10 +349,12 @@ class BaseDealImage(object):
             "***************all_original_images*********************",
             all_original_images,
         )
-        is_flip_800image = settings.getSysConfigs(
-            "basic_configs", "is_flip_800image", 1
-        )
-        image_deal_mode = int(is_flip_800image)
+        # is_flip_800image = settings.getSysConfigs(
+        #     "basic_configs", "is_flip_800image", 1
+        # )
+        # image_deal_mode = int(is_flip_800image)
+        # 服装固定写死  不执行检测反转
+        image_deal_mode = 0
         for image_dict in all_original_images:
             if windows:
                 if windows.state != 1:

+ 9 - 4
python/service/online_request/module_online_data.py

@@ -296,20 +296,25 @@ class AIGCDataRequest(object):
         if len(generate_ids) == 0:
             raise UnicornException("模特图生成失败")
         generate_id = generate_ids[0]
-        search_times = 60
+        search_times = 80
         status = 0
         result_image = None
         while search_times > 0:
             print(f"查询第{search_times}次")
             status, result_image = self.searchProgress(generate_id)
-            if status in [-1, 2]:
+            # status: -1=失败, 0=排队中, 1=进行中, 2=完成
+            if status == 2:  # 完成
+                break
+            if status == -1:  # 失败
                 break
+            # status为0(排队中)或1(进行中)时继续查询
             time.sleep(1)
             search_times -= 1
-        if not result_image:
+        # 循环结束后检查最终状态
+        if status == -1 or (status != 2 and search_times <= 0):
             raise UnicornException("模特图生成失败")
         save_image_path = download_image_with_pil(result_image, save_path)
-        print("上脚图save_image_path", result_image, save_image_path)
+        print("上脚图save_image_path",result_image, save_image_path)
         return save_image_path
 
     def generateProductSceneQW(self, local_path, prompt, save_path):

+ 1 - 1
python/settings.py

@@ -21,7 +21,7 @@ from databases import (
     batch_insert_device_configsNew,
 )
 # 追加配置参数 machine_type 拍照机设备类型;0鞋;1服装
-MACHINE_TYPE = 0
+MACHINE_TYPE = 1
 # 初始化数据表
 create_all_database()
 session = SqlQuery()