chore: continued working on #36

This commit is contained in:
Andre Henriques 2023-10-10 15:56:00 +01:00
parent ef1348fe93
commit 998a583bdb
2 changed files with 8 additions and 1 deletions

View File

@ -159,6 +159,7 @@ func trainDefinition(c *Context, model *BaseModel, definition_id string) (accura
"DataDir": path.Join(getDir(), "savedData", model.Id, "data"), "DataDir": path.Join(getDir(), "savedData", model.Id, "data"),
"RunPath": run_path, "RunPath": run_path,
"ColorMode": model.ImageMode, "ColorMode": model.ImageMode,
"Model": model,
}); err != nil { }); err != nil {
return return
} }
@ -166,7 +167,7 @@ func trainDefinition(c *Context, model *BaseModel, definition_id string) (accura
// Run the command // Run the command
out, err := exec.Command("bash", "-c", fmt.Sprintf("cd %s && python run.py", run_path)).Output() out, err := exec.Command("bash", "-c", fmt.Sprintf("cd %s && python run.py", run_path)).Output()
if err != nil { if err != nil {
fmt.Println(string(out)) c.Logger.Debug(string(out))
return return
} }

View File

@ -32,7 +32,13 @@ def pathToLabel(path):
#return tf.strings.as_string([path]) #return tf.strings.as_string([path])
def decode_image(img): def decode_image(img):
{{ if eq .Model.Format "png" }}
img = tf.io.decode_png(img, channels={{.ColorMode}}) 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) return tf.image.resize(img, image_size)
def process_path(path): def process_path(path):