+
+
+
+
+
+
+ + + 通用图像分割任务 +
+ +
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ 支持中英文，后端会自动翻译。多个对象可用逗号分隔。
+
+
+
+
+
+
+ 0.0
+ 1.0
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ {{ segmentResult.status === 'success' ? '分割成功' : '分割失败' }}
+{{ segmentResult.message }}
+
+
+
+
+ + + 可视化结果 +
+
+ ![]()
+
+
+
+
+
+
+
+
+ ![]()
+
+
+
+
+
+ {{ (seg.score * 100).toFixed(0) }}%
+
+
+
+ {{ seg.label }}
+
+
+ + + JSON 结果 +
+{{ JSON.stringify(segmentResult, null, 2) }}
+
@@ -972,6 +1122,21 @@
const tarotResult = ref(null);
const isRecognizing = ref(false);
+ // Segment State
+ const segmentFile = ref(null);
+ const segmentImageUrl = ref('');
+ const segmentPreview = ref(null);
+ const segmentPrompt = ref('');
+ const segmentConfidence = ref(0.7);
+ const segmentOptions = ref({
+ save_segment_images: false,
+ cutout: false,
+ perspective_correction: false,
+ highlight: false
+ });
+ const segmentResult = ref(null);
+ const isSegmenting = ref(false);
+
// Filters
const selectedTimeRange = ref('all');
const selectedType = ref('all');
@@ -1379,17 +1544,11 @@
}
formData.append('expected_count', tarotExpectedCount.value);
- // Use axios directly or a helper. Need to handle API Key if required by backend,
- // but admin usually has session. Wait, the backend endpoints like /recognize_tarot
- // require X-API-Key header.
- // The admin page uses cookie for /admin/api/* but /recognize_tarot is a public API protected by Key.
- // We should add the key to the header.
-
const config = {
headers: {
- 'X-API-Key': '123quant-speed' // Hardcoded as per fastAPI_tarot.py VALID_API_KEY
+ 'X-API-Key': '123quant-speed'
},
- timeout: 120000 // 2分钟超时，大模型响应较慢
+ timeout: 120000
};
const res = await axios.post('/recognize_tarot', formData, config);
@@ -1399,7 +1558,6 @@
console.error(e);
let msg = e.response?.data?.detail || e.message || '识别请求失败';
- // 针对 504 Gateway Timeout 或 请求超时做特殊提示
if (e.response && e.response.status === 504) {
msg = '请求超时 (504)：大模型处理时间较长。后台可能仍在运行，请稍后在“识别记录”中刷新查看结果。';
} else if (e.code === 'ECONNABORTED') {
@@ -1415,6 +1573,85 @@
}
};
+ // --- Segment Actions ---
+ const handleSegmentFileChange = (event) => {
+ const file = event.target.files[0];
+ if (file) {
+ segmentFile.value = file;
+ segmentImageUrl.value = '';
+ segmentPreview.value = URL.createObjectURL(file);
+ segmentResult.value = null;
+ }
+ };
+
+ const handleSegmentUrlInput = () => {
+ if (segmentImageUrl.value) {
+ segmentFile.value = null;
+ segmentPreview.value = segmentImageUrl.value;
+ segmentResult.value = null;
+ } else {
+ segmentPreview.value = null;
+ }
+ };
+
+ const clearSegmentInput = () => {
+ segmentFile.value = null;
+ segmentImageUrl.value = '';
+ segmentPreview.value = null;
+ segmentResult.value = null;
+ const fileInput = document.getElementById('dropzone-segment-file');
+ if (fileInput) fileInput.value = '';
+ };
+
+ const performSegment = async () => {
+ if ((!segmentFile.value && !segmentImageUrl.value) || !segmentPrompt.value) return;
+
+ isSegmenting.value = true;
+ segmentResult.value = null;
+
+ try {
+ const formData = new FormData();
+ if (segmentFile.value) {
+ formData.append('file', segmentFile.value);
+ } else {
+ formData.append('image_url', segmentImageUrl.value);
+ }
+ formData.append('prompt', segmentPrompt.value);
+ formData.append('confidence', segmentConfidence.value);
+ formData.append('save_segment_images', segmentOptions.value.save_segment_images);
+ formData.append('cutout', segmentOptions.value.cutout);
+ formData.append('perspective_correction', segmentOptions.value.perspective_correction);
+ formData.append('highlight', segmentOptions.value.highlight);
+
+ const config = {
+ headers: {
+ 'X-API-Key': '123quant-speed'
+ },
+ timeout: 120000
+ };
+
+ const res = await axios.post('/segment', formData, config);
+ segmentResult.value = res.data;
+
+ } catch (e) {
+ console.error(e);
+ let msg = e.response?.data?.detail || e.message || '分割请求失败';
+
+ if (e.response && e.response.status === 504) {
+ msg = '请求超时 (504)：处理时间较长。';
+ } else if (e.code === 'ECONNABORTED') {
+ msg = '请求超时：网络连接中断或服务器响应过慢。';
+ }
+
+ segmentResult.value = {
+ status: 'failed',
+ message: msg
+ };
+ } finally {
+ isSegmenting.value = false;
+ }
+ };
+
// --- Navigation & Helpers ---
const switchTab = (tab) => {
const prevTab = currentTab.value;
@@ -1506,6 +1743,7 @@
const map = {
'dashboard': '数据看板',
'tarot': '塔罗牌识别',
+ 'segment': '通用分割',
'history': '识别记录',
'files': '文件资源管理',
'prompts': '提示词工程',
@@ -1519,6 +1757,7 @@
const map = {
'dashboard': '系统运行状态与核心指标概览',
'tarot': 'SAM3 + Qwen-VL 联合识别与分割',
+ 'segment': '基于文本提示的通用图像分割 (Grounded SAM)',
'history': '所有视觉识别任务的历史流水',
'files': '查看和管理生成的图像及JSON结果',
'prompts': '调整各个识别场景的 System Prompt',
@@ -1734,7 +1973,10 @@
gpuUtilChartRef, gpuTempChartRef,
// Tarot
tarotFile, tarotImageUrl, tarotExpectedCount, tarotPreview, tarotResult, isRecognizing,
- handleTarotFileChange, handleUrlInput, clearTarotInput, recognizeTarot
+ handleTarotFileChange, handleUrlInput, clearTarotInput, recognizeTarot,
+ // Segment
+ segmentFile, segmentImageUrl, segmentPreview, segmentPrompt, segmentConfidence, segmentOptions, segmentResult, isSegmenting,
+ handleSegmentFileChange, handleSegmentUrlInput, clearSegmentInput, performSegment
};
}
}).mount('#app');