chore: continued working on #36
This commit is contained in:
parent
ef1348fe93
commit
998a583bdb
@ -159,6 +159,7 @@ func trainDefinition(c *Context, model *BaseModel, definition_id string) (accura
|
||||
"DataDir": path.Join(getDir(), "savedData", model.Id, "data"),
|
||||
"RunPath": run_path,
|
||||
"ColorMode": model.ImageMode,
|
||||
"Model": model,
|
||||
}); err != nil {
|
||||
return
|
||||
}
|
||||
@ -166,7 +167,7 @@ func trainDefinition(c *Context, model *BaseModel, definition_id string) (accura
|
||||
// Run the command
|
||||
out, err := exec.Command("bash", "-c", fmt.Sprintf("cd %s && python run.py", run_path)).Output()
|
||||
if err != nil {
|
||||
fmt.Println(string(out))
|
||||
c.Logger.Debug(string(out))
|
||||
return
|
||||
}
|
||||
|
||||
|
@ -32,7 +32,13 @@ def pathToLabel(path):
|
||||
#return tf.strings.as_string([path])
|
||||
|
||||
def decode_image(img):
|
||||
{{ if eq .Model.Format "png" }}
|
||||
img = tf.io.decode_png(img, channels={{.ColorMode}})
|
||||
{{ else if eq .Model.Format "jpeg" }}
|
||||
img = tf.io.decode_jpeg(img, channels={{.ColorMode}})
|
||||
{{ else }}
|
||||
ERROR
|
||||
{{ end }}
|
||||
return tf.image.resize(img, image_size)
|
||||
|
||||
def process_path(path):
|
||||
|
Loading…
Reference in New Issue
Block a user