Skip to content

Commit

Permalink
Move mobilenet model definitions to discojs/src/models folder
Browse files Browse the repository at this point in the history
  • Loading branch information
JulienVig authored and tharvik committed May 31, 2024
1 parent a4d6479 commit df0e389
Show file tree
Hide file tree
Showing 6 changed files with 14 additions and 11 deletions.
5 changes: 3 additions & 2 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
# dependencies
/node_modules/
# disco models
**/models
# model.json files
server/models
cli/models
# built
dist/

Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
import * as tf from '@tensorflow/tfjs'

import type { Model, Task, TaskProvider } from '../../index.js'
import { data, models } from '../../index.js'
import type { Model, Task, TaskProvider } from '../index.js'
import { data, models } from '../index.js'

import baseModel from './model.js'
import baseModel from '../models/mobileNet_v1_025_224.js'

export const cifar10: TaskProvider = {
getTask (): Task {
Expand Down
4 changes: 2 additions & 2 deletions discojs/src/default_tasks/index.ts
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
export { cifar10 } from './cifar10/index.js'
export { cifar10 } from './cifar10.js'
export { lusCovid } from './lus_covid.js'
export { skinMnist } from './skin_mnist.js'
export { mnist } from './mnist.js'
export { simpleFace } from './simple_face/index.js'
export { simpleFace } from './simple_face.js'
export { titanic } from './titanic.js'
export { wikitext } from './wikitext.js'
Original file line number Diff line number Diff line change
@@ -1,9 +1,8 @@
import * as tf from '@tensorflow/tfjs'

import type { Model, Task, TaskProvider } from '../../index.js'
import { data, models } from '../../index.js'

import baseModel from './model.js'
import type { Model, Task, TaskProvider } from '../index.js'
import { data, models } from '../index.js'
import baseModel from '../models/mobileNetV2_35_alpha_2_classes.js'

export const simpleFace: TaskProvider = {
getTask (): Task {
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
// Source: https://storage.googleapis.com/deai-313515.appspot.com/models/mobileNetV2_35_alpha_2_classes/model.json
// This model was converted using the tensorflow.js converter
export default {
format: "layers-model",
generatedBy: "keras v2.6.0",
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
// Source: https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_0.25_224/model.json
export default {
modelTopology: {
keras_version: "2.1.4",
Expand Down

0 comments on commit df0e389

Please sign in to comment.