我正在嘗試使用 MobileNet 創(chuàng)建一個簡單的圖像分類示例。我試圖讓它與react-dropzone 一起工作,在這種情況下,我最終需要將圖像轉(zhuǎn)換為Fileto的類型Tensor才能將其傳遞給model.classify.我也嘗試使用,fromPixels但為此,我必須將圖像轉(zhuǎn)換為ImageData.export default function ImageClassification() { const [isModelLoaded, setModelLoaded] = useState(false); const [uploadedFile, setUploadedFile] = useState(); const [classifier, setClassifier] = useState(); useEffect(() => { async function modelReady() { console.log("Not loaded:" + isModelLoaded); setClassifier( await MobileNet.load().then(model => { setModelLoaded(true); return model; }) ); } modelReady(); }, []); function onDrop(acceptedFiles: File[]) { console.log(acceptedFiles); setUploadedFile(acceptedFiles); console.log("After setting"); } function prepareImage(inputFile: File) { // How to convert inputFile to Tensor??? return image; } const { getRootProps, getInputProps, isDragActive } = useDropzone({ onDrop }); return ( <React.Fragment> {!isModelLoaded ? ( <CircularProgress /> ) : ( <div {...getRootProps()}> <input {...getInputProps()} /> {isDragActive ? ( <p>Drop the files here.. </p> ) : ( <p>Drag 'n' drop some files here, or click to select files</p> )} {uploadedFile && uploadedFile.map((item: File) => { const input = prepareImage(item); classifier.classify(input); return <h1>{item.name}</h1>; })} </div> )} </React.Fragment> );}任何有關(guān)這些的幫助將不勝感激。
如何在 JavaScript 中將文件轉(zhuǎn)換為張量
守著星空守著你
2021-12-23 15:54:07