Exemplo n.º 1
0
func Main() {
	file, err := os.Open("in.ppm")
	if err != nil {
		log.Fatal(err)
	}

	img, format, err := image.Decode(file)

	log.Println("Format: ", format)

	if err != nil {
		log.Fatal(err)
	}

	outFile, err := os.Create("out.ppm")
	if err != nil {
		log.Fatal(err)
	}

	err = pnm.Encode(outFile, img, pnm.PPM)
	if err != nil {
		log.Fatal(err)
	}

}
Exemplo n.º 2
0
func Example() {
	file, err := os.Open("in.ppm")
	if err != nil {
		log.Fatal(err)
	}

	img, err := image.Decode(file)

	if err != nil {
		log.Fatal(err)
	}

	outFile, err := os.Create("out.ppm")
	if err != nil {
		log.Fatal(err)
	}

	err = pnm.Encode(outFile, img, pnm.PPM)
	if err != nil {
		log.Fatal(err)
	}

}
Exemplo n.º 3
0
func main() {
	flag.Parse()
	fmt.Printf("Computing optical flow betwen %s and %s, result will be saved in %s and %s\n", finame1, finame2, magImageName, dirImageName)

	fin1, err := os.Open(finame1)
	if err != nil {
		log.Fatal(err)
	}
	defer fin1.Close()

	img1, _, err := image.Decode(fin1)
	if err != nil {
		log.Fatal(err)
	}

	fin2, err := os.Open(finame2)
	if err != nil {
		log.Fatal(err)
	}
	defer fin2.Close()

	img2, _, err := image.Decode(fin2)
	if err != nil {
		log.Fatal(err)
	}

	if !img1.Bounds().Eq(img2.Bounds()) {
		log.Fatal("The image bounds need to match")
	}

	// Create Gray float based images with overlap for mirroring boundaries
	f1 := floatimage.GrayFloatWithDummiesFromImage(img1)
	f2 := floatimage.GrayFloatWithDummiesFromImage(img2)

	min1, max1, mean1, var1 := analyse(f1)
	min2, max2, mean2, var2 := analyse(f2)
	fmt.Printf("min1 = %f, max1 = %f, mean1 = %f, var1 = %f\n", min1, max1, mean1, var1)
	fmt.Printf("min2 = %f, max2 = %f, mean2 = %f, var2 = %f\n", min2, max2, mean2, var2)

	uv := algorithms.OpticFlowHornSchunk(f1, f2, float32(alpha), iterations)
	magImg := algorithms.MagImage(uv)

	fout, err := os.Create(magImageName)
	if err != nil {
		log.Fatal(err)
	}
	defer fout.Close()
	magImg.ScaleToUnsignedByte()
	mag := magImg.Dedummify()

	err = pnm.Encode(fout, mag, pnm.PGM)
	if err != nil {
		log.Fatal(err)
	}

	uv.ColorFunc = func(x, y int, d []float32) color.Color {
		const mult = 100.0
		g := magImg.AtF(x, y)[0]
		return color.YCbCr{floatimage.Tu8c(g), floatimage.Tu8c(d[0]*mult + 127.5), floatimage.Tu8c(d[1]*mult + 127.5)}
	}

	uv.ColorModelFunc = func() color.Model {
		return color.YCbCrModel
	}

	fout2, err := os.Create(dirImageName)
	if err != nil {
		log.Fatal(err)
	}
	defer fout2.Close()

	err = pnm.Encode(fout2, uv.Dedummify(), pnm.PPM)
	if err != nil {
		log.Fatal(err)
	}
}