feat: closes #23

This commit is contained in:
Andre Henriques 2023-10-03 18:49:36 +01:00
parent 884a978e3b
commit f68f0e2444
4 changed files with 24 additions and 10 deletions

View File

@ -7,7 +7,6 @@ import (
"net/http"
"os"
"path"
"strconv"
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/utils"
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
@ -131,6 +130,8 @@ func handleRun(handle *Handle) {
vi := 0
var predictions = results[0].Value().([][]float32)[0]
fmt.Println(predictions)
for i, v := range predictions {
if v > vmax {
vi = i
@ -140,9 +141,24 @@ func handleRun(handle *Handle) {
os.RemoveAll(run_path)
rows, err := handle.Db.Query("select name from model_classes where model_id=$1 and class_order=$2;", model.Id, vi)
if err != nil { return Error500(err) }
if !rows.Next() {
LoadDefineTemplate(w, "/models/edit.html", "run-model-card", c.AddMap(AnyMap{
"Model": model,
"NotFound": true,
"Result": nil,
}))
return nil
}
var name string
if err = rows.Scan(&name); err != nil { return nil }
LoadDefineTemplate(w, "/models/edit.html", "run-model-card", c.AddMap(AnyMap{
"Model": model,
"Result": strconv.Itoa(vi),
"Result": name,
}))
return nil
})

View File

@ -83,12 +83,6 @@ func generateCvs(handle *Handle, run_path string, model_id string) (count int,
if err != nil { return }
defer data.Close()
type row struct {
path string
class_order int
}
f, err := os.Create(path.Join(run_path, "train.csv"))
if err != nil { return }
defer f.Close()

View File

@ -294,7 +294,11 @@
{{ define "run-model-card" }}
<form hx-headers='{"REQUEST-TYPE": "html"}' enctype="multipart/form-data" hx-post="/models/run" hx-swap="outerHTML">
{{ if .Result }}
{{ if .NotFound }}
<div>
The class was not found
</div>
{{ else if .Result }}
<div>
Img Class: {{.Result}}
</div>

View File

@ -34,7 +34,7 @@ def pathToLabel(path):
def decode_image(img):
# channels were reduced to 1 since image is grayscale
# TODO chnage channel number based if grayscale
img = tf.io.decode_png(img, channels=1)
img = tf.io.decode_png(img, channels=3)
return tf.image.resize(img, image_size)