Compare commits

2 Commits
Author SHA1 Message Date
539943419 f341159f1a ci(workflow): 更新 GitHub Actions 发布配置和构建脚本
更新 action-gh-release 从 v1 到 v2
优化构建脚本,增加错误处理和命令检查
使用更安全的脚本编写方式,避免全局变量污染
改进跨平台构建的兼容性和可维护性
2025-08-22 01:54:01 +08:00
539943419 97276224ee feat(plugins): 添加ONNX模型演示插件并优化组件加载逻辑
refactor(component2plugin): 改进组件加载逻辑避免重复加载
style(components.d.ts): 清理未使用的组件类型声明
chore(package.json): 添加build:normal脚本
2025-08-22 01:20:48 +08:00
10 changed files with 434 additions and 53 deletions

No files matched your search

+1 -1
View File
@@ -71,7 +71,7 @@ jobs:
- name: 创建自动发布 - name: 创建自动发布
if: github.event_name == 'push' && (github.ref == 'refs/heads/main' || github.ref == 'refs/heads/master') if: github.event_name == 'push' && (github.ref == 'refs/heads/main' || github.ref == 'refs/heads/master')
uses: softprops/action-gh-release@v1 uses: softprops/action-gh-release@v2
with: with:
name: Auto-Build-${{ steps.version.outputs.date }}-${{ steps.version.outputs.sha_short }} name: Auto-Build-${{ steps.version.outputs.date }}-${{ steps.version.outputs.sha_short }}
tag_name: auto-${{ steps.version.outputs.date }}-${{ steps.version.outputs.sha_short }} tag_name: auto-${{ steps.version.outputs.date }}-${{ steps.version.outputs.sha_short }}
+55 -38
View File
@@ -1,4 +1,7 @@
#!/bin/bash #!/bin/bash
set -Eeuo pipefail
IFS=$'\n\t'
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# 解析命令行参数 # 解析命令行参数
FULL_BUILD=false FULL_BUILD=false
@@ -25,6 +28,12 @@ check_upx() {
return 1 return 1
fi fi
} }
require_cmd() {
if ! command -v "$1" >/dev/null 2>&1; then
echo "$1 未找到,请安装后再试"
exit 1
fi
}
# 清理已有构建产物 # 清理已有构建产物
echo "清理已有构建产物..." echo "清理已有构建产物..."
@@ -94,14 +103,18 @@ if [ "$FULL_BUILD" = true ]; then
fi fi
echo "构建前端..." echo "构建前端..."
require_cmd pnpm
require_cmd go
pnpm run build pnpm run build
echo "复制前端文件到服务端..." echo "复制前端文件到服务端..."
rm -rf ./src-server/web/* rm -rf ./src-server/web/*
mkdir -p ./src-server/web/ mkdir -p ./src-server/web/
cp -r ./dist/* ./src-server/web/ if compgen -G "./dist/*" > /dev/null; then
cp -r ./dist/* ./src-server/web/
fi
cd ./src-server cd "$SCRIPT_DIR/src-server"
# ===== 构建阶段 ===== # ===== 构建阶段 =====
echo "===== 开始构建阶段 =====" echo "===== 开始构建阶段 ====="
@@ -111,20 +124,16 @@ echo "构建基础版本..."
# Windows AMD64 无GUI # Windows AMD64 无GUI
echo "构建 Windows AMD64 无GUI 版本" echo "构建 Windows AMD64 无GUI 版本"
go env -w GOOS=windows GOARCH=amd64 GOOS=windows GOARCH=amd64 go build -ldflags="-s -w" -o "$SCRIPT_DIR/miao-directory-amd64-win.exe"
go build -ldflags="-s -w" -o ../miao-directory-amd64-win.exe
# Linux AMD64 无GUI # Linux AMD64 无GUI
echo "构建 Linux AMD64 无GUI 版本" echo "构建 Linux AMD64 无GUI 版本"
go env -w GOOS=linux GOARCH=amd64 CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -ldflags="-s -w" -o "$SCRIPT_DIR/miao-directory-amd64-linux"
export CGO_ENABLED=0 chmod +x "$SCRIPT_DIR/miao-directory-amd64-linux"
go build -ldflags="-s -w" -o ../miao-directory-amd64-linux
export CGO_ENABLED=1
# Windows AMD64 GUI # Windows AMD64 GUI
echo "构建 Windows AMD64 GUI 版本" echo "构建 Windows AMD64 GUI 版本"
go env -w GOOS=windows GOARCH=amd64 GOOS=windows GOARCH=amd64 go build -tags webview -ldflags="-s -w -H=windowsgui" -o "$SCRIPT_DIR/miao-directory-amd64-win.gui.exe"
go build -tags webview -ldflags="-s -w -H=windowsgui" -o ../miao-directory-amd64-win.gui.exe
# 完整构建 - 额外的系统和架构 # 完整构建 - 额外的系统和架构
if [ "$FULL_BUILD" = true ]; then if [ "$FULL_BUILD" = true ]; then
@@ -136,25 +145,31 @@ if [ "$FULL_BUILD" = true ]; then
mv resource.syso resource.syso.bak mv resource.syso resource.syso.bak
fi fi
# 无需GUI和syso的构建批次 : # CGO 在每次构建调用中禁用,无需全局 export
export CGO_ENABLED=0
# 定义操作系统和架构映射 # 定义操作系统和架构映射
declare -A OS_ARCHS # 在工具函数处增加大小写格式化,避免使用 Bash 4 的 ${var^}/${var^^}
OS_ARCHS["linux"]="arm64 386 mips mips64 mips64le ppc64 ppc64le riscv64 s390x" uc_first() {
OS_ARCHS["darwin"]="amd64 arm64" local s="$1"
OS_ARCHS["windows"]="arm64 386" local first="${s:0:1}"
OS_ARCHS["freebsd"]="amd64 arm64 386" local rest="${s:1}"
OS_ARCHS["openbsd"]="amd64 arm64 386" printf "%s%s" "$(printf "%s" "$first" | tr '[:lower:]' '[:upper:]')" "$rest"
# OS_ARCHS["netbsd"]="amd64 arm64 386" }
# 遍历操作系统和架构进行构建 to_upper() {
for os in "${!OS_ARCHS[@]}"; do printf "%s" "$1" | tr '[:lower:]' '[:upper:]'
echo "构建 ${os^} 系列版本..." }
go env -w GOOS=$os OS_LIST="linux darwin windows freebsd openbsd"
for os in $OS_LIST; do
# 获取当前操作系统支持的架构列表 echo "构建 $(uc_first "$os") 系列版本..."
archs=${OS_ARCHS[$os]} case "$os" in
linux) archs="arm64 386 mips mips64 mips64le ppc64 ppc64le riscv64 s390x" ;;
darwin) archs="amd64 arm64" ;;
windows) archs="arm64 386" ;;
freebsd) archs="amd64 arm64 386" ;;
openbsd) archs="amd64 arm64 386" ;;
*) archs="" ;;
esac
for arch in $archs; do for arch in $archs; do
# 设置输出文件名后缀 # 设置输出文件名后缀
@@ -163,17 +178,19 @@ if [ "$FULL_BUILD" = true ]; then
suffix=".exe" suffix=".exe"
fi fi
echo "构建 ${os^} ${arch^^} 版本" echo "构建 $(uc_first "$os") $(to_upper "$arch") 版本"
go env -w GOARCH=$arch
# 根据条件设置不同的输出文件名格式 # 根据条件设置不同的输出文件名格式
if [[ "$os" == "darwin" || ("$os" == "linux" && "$arch" == "arm64") || ("$os" == "windows" && "$arch" == "arm64") ]]; then if [[ "$os" == "darwin" || ("$os" == "linux" && "$arch" == "arm64") || ("$os" == "windows" && "$arch" == "arm64") ]]; then
output_file="../miao-directory-${arch}-${os}${suffix}" output_file="$SCRIPT_DIR/miao-directory-${arch}-${os}${suffix}"
else else
output_file="../miao-directory-x-${arch}-${os}${suffix}" output_file="$SCRIPT_DIR/miao-directory-x-${arch}-${os}${suffix}"
fi fi
go build -ldflags="-s -w" -o "$output_file" CGO_ENABLED=0 GOOS="$os" GOARCH="$arch" go build -ldflags="-s -w" -o "$output_file"
if [ "$os" != "windows" ]; then
chmod +x "$output_file"
fi
done done
done done
@@ -183,7 +200,7 @@ if [ "$FULL_BUILD" = true ]; then
mv resource.syso.bak resource.syso mv resource.syso.bak resource.syso
fi fi
export CGO_ENABLED=1 : # 无需恢复 CGO,全程使用内联变量
fi fi
# ===== 压缩阶段 ===== # ===== 压缩阶段 =====
@@ -207,15 +224,15 @@ if check_upx; then
# 压缩Windows无GUI版本 # 压缩Windows无GUI版本
echo "正在压缩Windows AMD64无GUI版本" echo "正在压缩Windows AMD64无GUI版本"
upx $COMPRESSION_LEVEL -o ../miao-directory-amd64-win.upx.exe ../miao-directory-amd64-win.exe upx $COMPRESSION_LEVEL -o "$SCRIPT_DIR/miao-directory-amd64-win.upx.exe" "$SCRIPT_DIR/miao-directory-amd64-win.exe"
# 压缩Linux AMD64版本 # 压缩Linux AMD64版本
echo "正在压缩Linux AMD64版本" echo "正在压缩Linux AMD64版本"
upx $COMPRESSION_LEVEL -o ../miao-directory-amd64-linux.upx ../miao-directory-amd64-linux upx $COMPRESSION_LEVEL -o "$SCRIPT_DIR/miao-directory-amd64-linux.upx" "$SCRIPT_DIR/miao-directory-amd64-linux"
# 压缩Windows GUI版本 # 压缩Windows GUI版本
echo "正在压缩Windows AMD64 GUI版本" echo "正在压缩Windows AMD64 GUI版本"
upx $COMPRESSION_LEVEL -o ../miao-directory-amd64-win.gui.upx.exe ../miao-directory-amd64-win.gui.exe upx $COMPRESSION_LEVEL -o "$SCRIPT_DIR/miao-directory-amd64-win.gui.upx.exe" "$SCRIPT_DIR/miao-directory-amd64-win.gui.exe"
# 压缩额外架构版本 # 压缩额外架构版本
if [ "$FULL_BUILD" = true ]; then if [ "$FULL_BUILD" = true ]; then
@@ -223,15 +240,15 @@ if check_upx; then
# 压缩Linux ARM64版本 # 压缩Linux ARM64版本
echo "正在压缩Linux ARM64版本" echo "正在压缩Linux ARM64版本"
upx $COMPRESSION_LEVEL -o ../miao-directory-arm64-linux.upx ../miao-directory-arm64-linux upx $COMPRESSION_LEVEL -o "$SCRIPT_DIR/miao-directory-arm64-linux.upx" "$SCRIPT_DIR/miao-directory-arm64-linux"
# # 压缩macOS AMD64版本 # # 压缩macOS AMD64版本
# echo "正在压缩macOS AMD64版本" # echo "正在压缩macOS AMD64版本"
# upx $COMPRESSION_LEVEL -o ../miao-directory-amd64-darwin.upx ../miao-directory-amd64-darwin # upx $COMPRESSION_LEVEL -o "$SCRIPT_DIR/miao-directory-amd64-darwin.upx" "$SCRIPT_DIR/miao-directory-amd64-darwin"
# # 压缩macOS ARM64版本 # # 压缩macOS ARM64版本
# echo "正在压缩macOS ARM64版本" # echo "正在压缩macOS ARM64版本"
# upx $COMPRESSION_LEVEL -o ../miao-directory-arm64-darwin.upx ../miao-directory-arm64-darwin # upx $COMPRESSION_LEVEL -o "$SCRIPT_DIR/miao-directory-arm64-darwin.upx" "$SCRIPT_DIR/miao-directory-arm64-darwin"
fi fi
echo "所有压缩完成" echo "所有压缩完成"
-3
View File
@@ -15,19 +15,16 @@ declare module 'vue' {
MiaoLazyDiv: typeof import('./src/components/miaoLazyDiv.vue')['default'] MiaoLazyDiv: typeof import('./src/components/miaoLazyDiv.vue')['default']
MiaoMask: typeof import('./src/components/miaoMask.vue')['default'] MiaoMask: typeof import('./src/components/miaoMask.vue')['default']
MiaoPopupInput: typeof import('./src/components/miaoPopupInput.vue')['default'] MiaoPopupInput: typeof import('./src/components/miaoPopupInput.vue')['default']
NAlert: typeof import('naive-ui')['NAlert']
NBreadcrumb: typeof import('naive-ui')['NBreadcrumb'] NBreadcrumb: typeof import('naive-ui')['NBreadcrumb']
NBreadcrumbItem: typeof import('naive-ui')['NBreadcrumbItem'] NBreadcrumbItem: typeof import('naive-ui')['NBreadcrumbItem']
NButton: typeof import('naive-ui')['NButton'] NButton: typeof import('naive-ui')['NButton']
NCard: typeof import('naive-ui')['NCard'] NCard: typeof import('naive-ui')['NCard']
NCheckbox: typeof import('naive-ui')['NCheckbox']
NDropdown: typeof import('naive-ui')['NDropdown'] NDropdown: typeof import('naive-ui')['NDropdown']
NEllipsis: typeof import('naive-ui')['NEllipsis'] NEllipsis: typeof import('naive-ui')['NEllipsis']
NEmpty: typeof import('naive-ui')['NEmpty'] NEmpty: typeof import('naive-ui')['NEmpty']
NGrid: typeof import('naive-ui')['NGrid'] NGrid: typeof import('naive-ui')['NGrid']
NGridItem: typeof import('naive-ui')['NGridItem'] NGridItem: typeof import('naive-ui')['NGridItem']
NIcon: typeof import('naive-ui')['NIcon'] NIcon: typeof import('naive-ui')['NIcon']
NInput: typeof import('naive-ui')['NInput']
NQrCode: typeof import('naive-ui')['NQrCode'] NQrCode: typeof import('naive-ui')['NQrCode']
NScrollbar: typeof import('naive-ui')['NScrollbar'] NScrollbar: typeof import('naive-ui')['NScrollbar']
NSpace: typeof import('naive-ui')['NSpace'] NSpace: typeof import('naive-ui')['NSpace']
+1
View File
@@ -9,6 +9,7 @@
"build": "vue-tsc -b && vite build", "build": "vue-tsc -b && vite build",
"build:auto": "pnpm run build && go env -w GOOS=windows GOARCH=amd64 && rm -rf ./src-server/web/* && cp -r ./dist/* ./src-server/web/ && cd ./src-server && go env -w GOOS=windows && go build -o ../miao-directory.exe", "build:auto": "pnpm run build && go env -w GOOS=windows GOARCH=amd64 && rm -rf ./src-server/web/* && cp -r ./dist/* ./src-server/web/ && cd ./src-server && go env -w GOOS=windows && go build -o ../miao-directory.exe",
"build:fast": "bash build.auto.release.sh -fast", "build:fast": "bash build.auto.release.sh -fast",
"build:normal": "bash build.auto.release.sh",
"preview": "vite preview", "preview": "vite preview",
"format": "prettier --write src/" "format": "prettier --write src/"
}, },
File renamed without changes.
+4 -3
View File
@@ -1,13 +1,14 @@
import { registerComponentOption } from '@/class/PluginCenter' import { registerComponentOption } from '@/class/PluginCenter'
import FileTray from '@vicons/ionicons5/es/FileTray' import FileTray from '@vicons/ionicons5/es/FileTray'
// import MiaoDirectory from '@/plugins/miaoDirectory/miaoDirectory.vue' import MiaoDirectory from '@/plugins/miaoDirectory/miaoDirectory.vue'
const pluginConfig: registerComponentOption = { const pluginConfig: registerComponentOption = {
key: 'miaoDirectory', key: 'miaoDirectory',
name: '文件夹', name: '文件夹',
icon: FileTray, icon: FileTray,
getComponent: async () => // getComponent: async () =>
(await import('@/plugins/miaoDirectory/miaoDirectory.vue')).default, // (await import('@/plugins/miaoDirectory/miaoDirectory.vue')).default,
getComponent: async () => MiaoDirectory,
filter: (vDirs, vFiles) => { filter: (vDirs, vFiles) => {
if (vFiles.length) { if (vFiles.length) {
return false return false
+51
View File
@@ -0,0 +1,51 @@
import { registerComponentOption } from '@/class/PluginCenter'
import VirtualDirectory, { VirtualFile } from '@/class/VirtualDirectory'
import { Analytics } from '@vicons/ionicons5'
const pluginConfig: registerComponentOption = {
key: '45632456',
name: 'ONNX DEMO BIRD',
icon: Analytics,
getComponent: async () =>
(await import('./onnx_demo.vue')).default,
filter: (_vDirs: VirtualDirectory[], vFiles: VirtualFile[]) => {
if (vFiles.length === 0) {
return false
}
// 检查是否存在.onnx文件
let hasOnnxFile = false
// 检查是否存在图片文件
let hasImageFile = false
const imageSuffixList = [
'xbm', 'tif', 'pjp', 'svgz', 'jpg', 'jpeg', 'ico', 'tiff', 'gif',
'svg', 'jfif', 'webp', 'png', 'bmp', 'pjpeg', 'avif'
]
for (let vf of vFiles) {
if (vf.name.endsWith('.onnx')) {
hasOnnxFile = true
return true
}
for (let end of imageSuffixList) {
if (vf.name.endsWith(end)) {
hasImageFile = true
break
}
}
if (hasOnnxFile && hasImageFile) {
return true
}
}
return false
}
}
export default pluginConfig
import { PluginExportType } from '@/types/type'
export const type = PluginExportType.component
+306
View File
@@ -0,0 +1,306 @@
<template>
<miao-drop-handler @on-virtual-files="handleDrop">
<miao-message-provider ref="miaoMessageRef">
<div class="onnx-demo" ref="rootRef">
<n-card title="ONNX模型演示" class="onnx-demo-card">
<n-grid :cols="2" :x-gap="12">
<n-grid-item>
<div class="onnx-demo-section">
<h3>输入图像</h3>
<div v-if="inputImage" class="onnx-demo-image-container">
<img :src="inputImage.url" class="onnx-demo-image" />
<p>{{ inputImage.name }}</p>
</div>
<n-empty v-else description="无图像" />
</div>
</n-grid-item>
<n-grid-item>
<div class="onnx-demo-section">
<h3>ONNX模型</h3>
<div v-if="onnxModel" class="onnx-demo-model-info">
<p>模型: {{ onnxModel.name }}</p>
<p>大小: {{ formatFileSize(onnxModel.size) }}</p>
</div>
<n-empty v-else description="无模型" />
</div>
</n-grid-item>
</n-grid>
<!-- <div class="onnx-demo-control">
<n-button type="primary" @click="processImage" :disabled="!inputImage || !onnxModel || isProcessing"
:loading="isProcessing">
处理图像
</n-button>
</div> -->
<div class="onnx-demo-result" v-if="modelOutput">
<h3>模型输出:</h3>
<n-card embedded>
<pre>{{ modelOutput }}</pre>
</n-card>
</div>
</n-card>
</div>
</miao-message-provider>
</miao-drop-handler>
</template>
<script setup lang="ts">
import miaoDropHandler from '@/components/miaoDropHandler.vue'
import miaoMessageProvider from '@/components/miaoAlertTipProvider.vue'
import { VirtualFile } from '@/class/VirtualDirectory'
import VirtualDirectory from '@/class/VirtualDirectory'
import { onMounted, ref, computed, watch } from 'vue'
// @ts-ignore
let ort: any;
const currentFiles = defineModel<VirtualFile[]>('currentFiles', {
required: true
})
const currentDirectories = defineModel<VirtualDirectory[]>(
'currentDirectories',
{
required: true
}
)
const miaoMessageRef = ref<InstanceType<typeof miaoMessageProvider>>()
const rootRef = ref()
const isProcessing = ref(false)
const modelOutput = ref<string | null>(null)
// 获取第一个图像文件
const inputImage = computed(() => {
const imageSuffixList = [
'xbm', 'tif', 'pjp', 'svgz', 'jpg', 'jpeg', 'ico', 'tiff', 'gif',
'svg', 'jfif', 'webp', 'png', 'bmp', 'pjpeg', 'avif'
]
for (const file of currentFiles.value) {
for (const suffix of imageSuffixList) {
if (file.name.toLowerCase().endsWith(suffix)) {
return file
}
}
}
return null
})
// 获取ONNX模型文件
const onnxModel = computed(() => {
for (const file of currentFiles.value) {
if (file.name.toLowerCase().endsWith('.onnx')) {
return file
}
}
return null
})
const handleDrop = (vFiles: VirtualFile[]) => {
const imageSuffixList = [
'xbm', 'tif', 'pjp', 'svgz', 'jpg', 'jpeg', 'ico', 'tiff', 'gif',
'svg', 'jfif', 'webp', 'png', 'bmp', 'pjpeg', 'avif'
]
let imgFile: VirtualFile | null = null
out: for (const file of vFiles) {
for (const suffix of imageSuffixList) {
if (file.name.toLowerCase().endsWith(suffix)) {
imgFile = file
break out
}
}
}
if (imgFile) {
currentFiles.value.unshift(imgFile)
}
}
// 格式化文件大小
const formatFileSize = (size: number): string => {
if (size < 1024) {
return `${size} B`
} else if (size < 1024 * 1024) {
return `${(size / 1024).toFixed(2)} KB`
} else {
return `${(size / (1024 * 1024)).toFixed(2)} MB`
}
}
const resizeImage = async (img: ImageBitmap, width: number, height: number): Promise<ImageBitmap> => {
const canvas = new OffscreenCanvas(width, height)
canvas.width = width
canvas.height = height
const ctx = canvas.getContext('2d')
ctx?.drawImage(img, 0, 0, width, height)
return await createImageBitmap(canvas)
}
// 处理图像
const processImage = async () => {
if (!inputImage.value || !onnxModel.value) {
miaoMessageRef.value?.alertTip('请选择图像和ONNX模型文件', { type: 'error', timeout: 2000 })
return
}
isProcessing.value = true
modelOutput.value = null
try {
const alertUpdate = miaoMessageRef.value!.alertTip('正在加载ONNX模型...', { type: 'info', timeout: 2000 })
const modelResponse = await fetch(onnxModel.value.url)
const modelBuffer = await modelResponse.arrayBuffer()
alertUpdate('正在创建推理会话...')
const session = await ort.InferenceSession.create(modelBuffer)
alertUpdate('正在加载图像...')
const imgResponse = await fetch(inputImage.value.url)
const imgBlob = await imgResponse.blob()
alertUpdate('正在预处理图像...')
const size = 448
const img = await createImageBitmap(imgBlob)
const data = await resizeImage(img, size, size)
const canvas = new OffscreenCanvas(size, size)
const ctx = canvas.getContext('2d') as unknown as CanvasRenderingContext2D
ctx.drawImage(data, 0, 0, size, size)
const imageData = ctx.getImageData(0, 0, size, size).data
const newImageData = new Uint8ClampedArray(size * size * 3)
for (let i = 0; i < size * size; i++) {
newImageData[i * 3] = imageData[i * 4]
newImageData[i * 3 + 1] = imageData[i * 4 + 1]
newImageData[i * 3 + 2] = imageData[i * 4 + 2]
}
// const greyScale = [];
// for (let i = 0; i < imageData.length; i += 4) {
// greyScale.push(
// (imageData[i] * 0.299 + imageData[i + 1] * 0.587 + imageData[i + 2] * 0.114 - 127.5) /
// 127.5
// );
// }
// debugger
const inputTensor = new ort.Tensor('float32', new Float32Array(newImageData), [1, 3, size, size])
alertUpdate('执行模型推理...')
const outputMap = await session.run({
input: inputTensor
})
alertUpdate('处理模型输出结果...')
const outputTensor = Object.values(outputMap)[0]
const formattedOutput = JSON.stringify(
{
data: [...[...(outputTensor as { data: Float32Array }).data].entries()].sort((a, b) => b[1] - a[1]).map(([index, value]) => ({ index, value }))
},
null,
2
)
modelOutput.value = formattedOutput
alertUpdate('处理完成!', { type: 'success' })
} catch (error) {
console.error('ONNX处理错误:', error)
miaoMessageRef.value?.alertTip(`处理出错`, { type: 'error', timeout: 2000 })
modelOutput.value = `错误: ${error}`
} finally {
isProcessing.value = false
}
}
onMounted(async () => {
console.log('currentDirectories.value[0].url', currentDirectories.value[0]?.url)
if (currentDirectories.value && currentDirectories.value.length > 0 && currentDirectories.value[0].url) {
const ortModule = await import(/* @vite-ignore */ `${currentDirectories.value[0].url}dist/ort.all.min.mjs`);
ort = ortModule;
console.log('ort', ort)
} else {
// @ts-ignore
const ortModule = await import(/* @vite-ignore */ 'https://cdn.jsdelivr.net/npm/onnxruntime-web/dist/ort.all.min.js');
ort = ortModule;
}
await initModel();
})
const initModel = async () => {
if (inputImage.value && onnxModel.value) {
miaoMessageRef.value?.alertTip('已检测到图像和ONNX模型,可以进行处理', { type: 'info', timeout: 2000 })
}
watch(() => inputImage.value, () => {
processImage()
}, {
immediate: true
})
}
</script>
<style scoped>
.onnx-demo {
width: 100%;
height: 100%;
padding: 16px;
overflow: auto;
}
.onnx-demo-card {
margin-bottom: 16px;
}
.onnx-demo-section {
padding: 8px;
text-align: center;
border: 1px dashed #ccc;
border-radius: 4px;
min-height: 200px;
display: flex;
flex-direction: column;
justify-content: center;
}
.onnx-demo-image-container {
display: flex;
flex-direction: column;
align-items: center;
}
.onnx-demo-image {
max-width: 100%;
max-height: 180px;
object-fit: contain;
}
.onnx-demo-model-info {
text-align: left;
width: 100%;
}
.onnx-demo-control {
display: flex;
justify-content: center;
margin: 16px 0;
}
.onnx-demo-result {
margin-top: 16px;
border-top: 1px solid #eee;
padding-top: 16px;
}
.onnx-demo-result pre {
white-space: pre-wrap;
word-break: break-all;
max-height: 300px;
overflow: auto;
background-color: #f5f5f5;
padding: 8px;
border-radius: 4px;
}
</style>
+16 -8
View File
@@ -3,6 +3,7 @@ import VirtualDirectory, { VirtualFile } from "@/class/VirtualDirectory"
import VirtualPage from "@/class/VirtualPage" import VirtualPage from "@/class/VirtualPage"
import usePluginCenter from "@/hooks/usePluginCenter" import usePluginCenter from "@/hooks/usePluginCenter"
import useVirtualPages from "@/hooks/useVirtualPages" import useVirtualPages from "@/hooks/useVirtualPages"
import { Component } from "vue"
export const componentOption2pluginOption = (option: registerComponentOption) => { export const componentOption2pluginOption = (option: registerComponentOption) => {
const { const {
@@ -17,7 +18,8 @@ export const componentOption2pluginOption = (option: registerComponentOption) =>
single = false, single = false,
priority = 0 priority = 0
} = option } = option
let _component: any let _component: Component
let _componentPromise: Promise<Component>
const views = useVirtualPages() const views = useVirtualPages()
const pluginCenter = usePluginCenter() const pluginCenter = usePluginCenter()
return { return {
@@ -36,13 +38,19 @@ export const componentOption2pluginOption = (option: registerComponentOption) =>
} }
if (_component === undefined) { if (_component === undefined) {
const setAlertTip = pluginCenter.globalAlertTip(`加载 ${name} 插件中`) if (_componentPromise !== undefined) {
_component = await getComponent() await _componentPromise
setAlertTip && }
setAlertTip(`加载 ${name} 插件完成`, { else {
type: 'success', const setAlertTip = pluginCenter.globalAlertTip(`加载 ${name} 插件中`)
timeout: 2000 _componentPromise = getComponent()
}) _component = await _componentPromise
setAlertTip &&
setAlertTip(`加载 ${name} 插件完成`, {
type: 'success',
timeout: 2000
})
}
} }
const _findRes = views.find(_component) as VirtualPage[] const _findRes = views.find(_component) as VirtualPage[]
if (single && _findRes.length !== 0) { if (single && _findRes.length !== 0) {