1
This commit is contained in:
@@ -96,20 +96,17 @@ type AdminDatasetUploadRes struct {
|
||||
}
|
||||
|
||||
// AdminDatasetGenerateReq AI 生成图片(同步执行;prompt 可手填覆盖模板,禁止含目标位置描述;
|
||||
// 未手填走 imageGen.promptTemplate 组装:species/scene 用表单选择,数量词=animalCount。
|
||||
// distanceMin/distanceMax(米)与 animalCount 填写后逐张校验:RF-DETR 检测目标框(去重口径),
|
||||
// 数量精确匹配、估算距离(物理公式)越界自动重新生成(上限 consts.GenValidateMaxAttempts 次)。
|
||||
// 文件名前缀为数据集属性(datasets 表 name_prefix),不再随生成请求传入
|
||||
// 未手填走 imageGen.promptTemplate 组装:species 用表单选择(scene/action/light 随机),数量词=animalCount。
|
||||
// animalCount 仅作为标注框数上限存储(自动标注按置信度裁剪 ≤N),不参与生成校验;
|
||||
// 距离校验已取消(2026-08-28)。文件名前缀为数据集属性(datasets 表 name_prefix)
|
||||
type AdminDatasetGenerateReq struct {
|
||||
g.Meta `path:"/datasets/generate" method:"post" summary:"AI 生成图片" tags:"管理端"`
|
||||
DatasetId int64 `json:"datasetId" v:"required|min:1" dc:"数据集 id"`
|
||||
Species string `json:"species" v:"length:0,50" dc:"动物名称(如家鸽/山斑鸠;留空按数据集名取物种池随机)"`
|
||||
Prompt string `json:"prompt" v:"length:0,500" dc:"完整提示词(可选,填写则覆盖模板;禁止含位置描述)"`
|
||||
Count int `json:"count" v:"required|integer|min:1|max:8" dc:"生成张数 1-8"`
|
||||
Size string `json:"size" v:"required|in:704x1248" dc:"图片尺寸"`
|
||||
DistanceMin int `json:"distanceMin" v:"integer|min:0" dc:"距离范围下限(米,0=不校验距离)"`
|
||||
DistanceMax int `json:"distanceMax" v:"integer|min:0" dc:"距离范围上限(米,0=不校验距离)"`
|
||||
AnimalCount int `json:"animalCount" v:"integer|min:0|max:20" dc:"画面目标数量(精确匹配,0=不校验;标注框数不超过该值)"`
|
||||
g.Meta `path:"/datasets/generate" method:"post" summary:"AI 生成图片" tags:"管理端"`
|
||||
DatasetId int64 `json:"datasetId" v:"required|min:1" dc:"数据集 id"`
|
||||
Species string `json:"species" v:"length:0,50" dc:"动物名称(如家鸽/山斑鸠;留空按数据集名取物种池随机)"`
|
||||
Prompt string `json:"prompt" v:"length:0,500" dc:"完整提示词(可选,填写则覆盖模板;禁止含位置描述)"`
|
||||
Count int `json:"count" v:"required|integer|min:1|max:8" dc:"生成张数 1-8"`
|
||||
Size string `json:"size" v:"required|in:704x1248" dc:"图片尺寸"`
|
||||
Distance int `json:"distance" v:"required|min:1" dc:"目标距离(米,固定值,注入提示词)"`
|
||||
}
|
||||
|
||||
// AdminDatasetGenOptionsReq 生成表单选项(物种池/场景池,来自 config imageGen 配置)
|
||||
@@ -124,7 +121,8 @@ type AdminDatasetGenOptionsRes struct {
|
||||
}
|
||||
|
||||
type AdminDatasetGenerateRes struct {
|
||||
Generated int `json:"generated"` // 成功张数(失败时可能 < count,已成功的图保留)
|
||||
Generated int `json:"generated"` // 成功入库张数(校验不合格的已跳过)
|
||||
Rejected int `json:"rejected"` // 连续 N 次校验不合格被跳过的张数
|
||||
}
|
||||
|
||||
// AdminDatasetImagesReq 数据集图片列表(标注工作台/网格预览用)
|
||||
|
||||
Reference in New Issue
Block a user