mirror of https://bitbucket.org/ausocean/av.git
131 lines
2.5 KiB
Go
131 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"gonum.org/v1/plot"
|
|
"gonum.org/v1/plot/plotter"
|
|
"gonum.org/v1/plot/plotutil"
|
|
"gonum.org/v1/plot/vg"
|
|
|
|
"gocv.io/x/gocv"
|
|
)
|
|
|
|
func main() {
|
|
n := 10
|
|
|
|
// Load template
|
|
template := gocv.IMRead("template.jpg", gocv.IMReadGrayScale)
|
|
|
|
// Read images
|
|
images := make([]gocv.Mat, n)
|
|
|
|
for i := 0; i < n; i++ {
|
|
images[i] = gocv.IMRead(fmt.Sprintf("images/generation-5/000%v.jpg", i), gocv.IMReadGrayScale)
|
|
}
|
|
|
|
// Calulate turbidity
|
|
ts := TurbiditySensor{template: template, k1: 90, k2: 90, scale: 5.0, alpha: 1.0}
|
|
|
|
err := ts.Evaluate(n, images)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}
|
|
|
|
// RESULTS
|
|
type Results struct {
|
|
turbidity []float64
|
|
saturation []float64
|
|
contrast []float64
|
|
}
|
|
|
|
func (r *Results) New(n int) {
|
|
r.turbidity = make([]float64, n)
|
|
r.saturation = make([]float64, n)
|
|
r.contrast = make([]float64, n)
|
|
}
|
|
|
|
func (r *Results) Append(saturation, contrast, turbidity float64, i int) {
|
|
r.saturation[i] = saturation
|
|
r.contrast[i] = contrast
|
|
r.turbidity[i] = turbidity
|
|
}
|
|
|
|
func Normalize(slice []float64, n int) []float64 {
|
|
|
|
max := -1e10
|
|
min := 1e10
|
|
|
|
out := make([]float64, n)
|
|
|
|
if n <= 1 {
|
|
return slice
|
|
}
|
|
|
|
for i := 0; i < n; i++ {
|
|
if slice[i] > max {
|
|
max = slice[i]
|
|
}
|
|
if slice[i] < min {
|
|
min = slice[i]
|
|
}
|
|
}
|
|
|
|
for i := 0; i < n; i++ {
|
|
out[i] = (slice[i] - min) / (max - min)
|
|
}
|
|
|
|
return out
|
|
}
|
|
|
|
// PLOTTING
|
|
|
|
func PlotResults(x, saturation, contrast []float64) error {
|
|
err := plotToFile(
|
|
"Results",
|
|
"Filter Size",
|
|
"Score",
|
|
func(p *plot.Plot) error {
|
|
return plotutil.AddLinePoints(p,
|
|
"AMEE", plotterXY(x, contrast),
|
|
"EME", plotterXY(x, saturation),
|
|
)
|
|
},
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("Could not plot results: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// plotToFile creates a plot with a specified name and x&y titles using the
|
|
// provided draw function, and then saves to a PNG file with filename of name.
|
|
func plotToFile(name, xTitle, yTitle string, draw func(*plot.Plot) error) error {
|
|
p := plot.New()
|
|
|
|
p.Title.Text = name
|
|
p.X.Label.Text = xTitle
|
|
p.Y.Label.Text = yTitle
|
|
|
|
err := draw(p)
|
|
if err != nil {
|
|
return fmt.Errorf("could not draw plot contents: %w", err)
|
|
}
|
|
|
|
if err := p.Save(15*vg.Centimeter, 15*vg.Centimeter, "plots/"+name+".png"); err != nil {
|
|
return fmt.Errorf("could not save plot: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// plotterXY provides a plotter.XYs type value based on the given x and y data.
|
|
func plotterXY(x, y []float64) plotter.XYs {
|
|
xy := make(plotter.XYs, len(x))
|
|
for i := range x {
|
|
xy[i].X = x[i]
|
|
xy[i].Y = y[i]
|
|
}
|
|
return xy
|
|
}
|