feat generate 迁移

app/core/config.py

@@ -19,59 +19,56 @@ class Settings(BaseSettings):
     LOGGING_CONFIG_FILE = os.path.join(BASE_DIR, 'logging_env.py')
 
 
+DEBUG = True
+ENV = 0
+if DEBUG:
+    LOGS_PATH = "logs/errors.log"
+else:
+    LOGS_PATH = "app/logs/errors.log"
+
+RABBITMQ_ENV = ""
+
+if ENV == 1:
+    RABBITMQ_ENV = "dev"
+elif ENV == 2:
+    RABBITMQ_ENV = "local"
+
 settings = Settings()
 
-ckpt = 'service/super_resolution_ccsr/weights/real-world_ccsr.ckpt'
-config = 'service/super_resolution_ccsr/configs/model/ccsr_stage2.yaml'
-steps = 45
-sr_scale = 4
-repeat_times = 1
-tiled = False
-tile_size = 512
-tile_stride = 256
-color_fix_type = "adain"
-t_max = 0.6667
-t_min = 0.3333
-show_lq = False
-skip_if_exist = False
-seed = 233
-device = "cuda"
-tile_diffusion = False  #
-tile_diffusion_size = 512
-tile_diffusion_stride = 256
-tile_vae = True
-vae_decoder_tile_size = 224
-vae_encoder_tile_size = 1024
-strength = 1
 # minio 配置
-sr_bucket = "test"
 MINIO_IP = "www.minio.aida.com.hk"
 MINIO_PORT = 9000
 MINIO_ACCESS = 'vXKFLSJkYeEq2DrSZvkB'
 MINIO_SECRET = 'uKTZT3x7C43WvPN9QTc99DiRkwddWZrG9Uh3JVlR'
+MINIO_SECURE = True
 
 # redis 配置
 REDIS_HOST = "10.1.1.240"
 REDIS_PORT = "6379"
 REDIS_DB = "2"
 
-MINIO_SECURE = True
-
-SR_MODEL_NAME = "super_resolution"
-SR_TRITON_URL = "10.1.1.240:10031"
-
 # rabbitmq config
-
 RABBITMQ_PARAMS = {
     "host": "18.167.251.121",
     "port": 5672,
     "credentials": pika.credentials.PlainCredentials(username='rabbit', password='123456'),
     "virtual_host": "/"
 }
-RABBITMQ_QUEUES = os.getenv("RABBITMQ_QUEUES", "SuperResolution-local")
 
-DEBUG = True
-if DEBUG:
-    LOGS_PATH = "logs/errors.log"
-else:
-    LOGS_PATH = "app/logs/errors.log"
+# SR service config
+SR_MODEL_NAME = "super_resolution"
+SR_TRITON_URL = "10.1.1.240:10031"
+SR_RABBITMQ_QUEUES = os.getenv("SR_RABBITMQ_QUEUES", "SuperResolution-local")
+
+# GenerateImage service config
+GI_MODEL_NAME = '_stable_diffusion'
+GI_MODEL_URL = '10.1.1.240:7001'
+GI_RABBITMQ_QUEUES = os.getenv("GI_RABBITMQ_QUEUES", f"GenerateImage-{RABBITMQ_ENV}")
+
+# SEG service config
+SEG_MODEL_URL = '10.1.1.240:10000'
+SEGMENTATION = {
+    "name": "seg_ocrnet_hr18",
+    "input": "seg_input__0",
+    "output": "seg_output__0",
+}