infinitered/nsfwjs

Supporting model load from indexeddb

Open

#430 opened on 2020年10月24日

GitHub で見る
 (2 comments) (0 reactions) (0 assignees)JavaScript (7,090 stars) (507 forks)batch import
Browser Issuesgood first issuehelp wanted

説明

https://github.com/infinitered/nsfwjs/blob/ebcd41c46087a3f42c6577f96acc53d7a934b068/src/index.ts#L68

Hello, it seems, although not explicit I can save the model to different schemas by referencing the underlying "model" attribute in the model returned by nsfwjs.load() e.g.

nsfwjs.load(path).then(function (newModel) { console.log("path", path); if(newModel) { newModel.model.save('indexeddb://' + SOME_KEY); } }).catch(error => { console.error('onRejected function called: ' + error.message); })

However, I am unable to then reload the model from say an indexeddb location using:

nsfwjs.load('indexeddb://' + SOME_KEY) as the line above in the code just checks for a string, assumes it's a url or relative path, rather than another schema location and them appends 'model.js'.

I think tf.js supports loading and saving to other schemas, it'd be cool if NSFWJS followed suit ...

コントリビューターガイド

技術スタック
javascripttypescript
領域
machine learningfrontend
Issue 種別
feature
難度新しい貢献者にとっての実装難度です。1 は非常に小さな変更、5 は専門的な作業です。
3
推定時間経験ある貢献者が調査、実装、テスト、pull request 準備にかけるおおよその時間範囲です。
half day
活動状況Issue が今どれだけ取り組みやすいかを示します。新しい、活発、古い、ブロック中、またはメンテナー入力待ちなどです。
stale
明確さ期待される変更、受け入れ条件、次の手順がどれだけ明確かを示します。
clear
前提条件
TensorFlow.js model APIJavaScript/TypeScript
初心者向け度初回貢献者にどれだけ取り組みやすいかを 1-100 で推定したスコアです。
65
調査方針
Examine src/index.ts line 68 where the load function currently checks if the path is a string and assumes a URL or relative path, then appends 'model.json'. The fix should detect other schemes like 'indexeddb://' and use the appropriate TensorFlow.js loader. Look at TF.js documentation for tf.loadLayersModel() which supports various URL schemes. Modify the load function to handle non HTTP schemes by passing the path directly to tf.loadLayersModel without appending 'model.json' when a scheme is detected.
Supporting model load from indexeddb · infinitered/nsfwjs#430 | Good First Issue