SCHISANDRA/schisandra-cloud-album-front
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

✨ add anime pictures and real pictures to be intelligently classified

Browse Source
This commit is contained in:
landaiqing
2025-01-07 22:46:06 +08:00
parent 90a68221fe
commit fed52e66f9
25 changed files with 38 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/store/modules/uploadStore.ts
View File

@@ -4,7 +4,8 @@ import i18n from "@/locales";
import {NSFWJS} from "nsfwjs";
import {message} from "ant-design-vue";
import {loadCocoSsd, loadMobileNet} from "@/utils/tfjs";
// import {loadCocoSsd, loadMobileNet} from "@/utils/tfjs/tfjs.ts";
import {loadModel, predictImage} from "@/utils/tfjs/anime_classifier.ts";
export const useUploadStore = defineStore(
'upload',
@@ -25,11 +26,19 @@ export const useUploadStore = defineStore(
message.error(i18n.global.t('comment.illegalImage'));
return false;
}
const predictions = await loadMobileNet(image);
console.log(predictions);
// const predictions = await loadMobileNet(image);
// console.log(predictions);
//
// const prediction = await loadCocoSsd(image);
// console.log(prediction);
const prediction = await loadCocoSsd(image);
console.log(prediction);
const model = await loadModel('/tfjs/anime_classifier/model.json');
// 进行预测
const output = await predictImage(model, image);
console.log(output);
// console.log('Predicted Class:', predictedClass);
return true;
}