Ver código fonte

修改prompt,优化模型性能

Sherlock1011 2 meses atrás
pai
commit
3951f639db
1 arquivos alterados com 8 adições e 5 exclusões
  1. 8 5
      config/config.py

+ 8 - 5
config/config.py

@@ -46,8 +46,8 @@ PROMPT_EXTRACT_COMPONENTS = """
 PROMPT_EXTRACT_KEYWORD = """
 你是一个专业的化学品安全标签说明识别助手。
 请从图像中提取安全提醒信号词和危险性说明
-安全信号词:通常以比较醒目的方式显示,如"危险"、"警告"等
-危险性说明:通常在安全提醒词附近。
+安全信号词:通常以比较醒目的方式显示,只识别"危险"、"警告"
+危险性说明:通常在安全提醒词附近,只返回中文的表示
 
 按照以下JSON格式输出结果:
 {
@@ -61,7 +61,10 @@ PROMPT_EXTRACT_KEYWORD = """
 # 步骤4:提取防范说明
 PROMPT_EXTRACT_PREVENTION = """
 你是一个专业的化学品安全标签说明识别助手。
-请从图像中提取防范说明,包括:预防措施、事故响应、安全存储、废弃处置信息。
+请从图像中提取防范说明,包括:预防措施、事故响应、安全存储、废弃处置信息:
+注意,要精准定位到预防措施、事故响应、安全存储、废弃处置的位置一般都有【预防措施】、【事故响应】、【安全存储】、【废弃处置】的标识
+在定位之后,精准的提取对应信息的内容,禁止篡改、编造
+如果对应的内容有英文表示,只需要返回其中文表示,不需要返回英文表示
 
 按照以下JSON格式输出结果:
 {
@@ -153,12 +156,12 @@ PROMPT_EXTRACT_ICON = """
 
 按照以下JSON格式输出结果:
 {
-    "tag_images": ["GHS06", "GHS08", "GHS09"]
+    "tag_images": [识别到的形象图]
 }
 
 注意:
 1. 必须按照图像中象形图从左到右的实际顺序排列
-2. 如果某个位置的象形图无法识别,用空字符串""占位
+2. 如果某个位置的象形图无法识别则跳过
 3. 识别出的象形图用对应的GHS编号(如GHS01-GHS09)表示
 4. 返回结果必须是标准JSON格式,不要包含```json```标记
 """