main.go

package main

import (
	"fmt"
	"io/ioutil"
	"log"
	"math/rand"
	"net/http"
	"time"

	"github.com/marianina8/haar-training/images"
	"github.com/marianina8/haar-training/samples"
	"github.com/marianina8/haar-training/training"
)

var (
	client = &http.Client{
		Timeout: 10 * time.Second,
	}
)

func init() {
	rand.Seed(time.Now().Unix())
}

func main() {

	// CREATE COMMAND OPTION THAT TAKES FILEPATH TO POSITIVE FILE

	// 1. DOWNLOAD NEGATIVE BACKGROUND FILES

	links := []string{
		"http://www.image-net.org/api/text/imagenet.synset.geturls?wnid=n12102133",
		"http://www.image-net.org/api/text/imagenet.synset.geturls?wnid=n09436708",
		"http://www.image-net.org/api/text/imagenet.synset.geturls?wnid=n12992868",
		"http://www.image-net.org/api/text/imagenet.synset.geturls?wnid=n07942152",
		"http://www.image-net.org/api/text/imagenet.synset.geturls?wnid=n02913152",
		"http://www.image-net.org/api/text/imagenet.synset.geturls?wnid=n02913152",
		"http://www.image-net.org/api/text/imagenet.synset.geturls?wnid=n04105893",
		"http://www.image-net.org/api/text/imagenet.synset.geturls?wnid=n03089879",
		"http://www.image-net.org/api/text/imagenet.synset.geturls?wnid=n10529231",
	}
	images.Get(links, "negatives", true, 1, 4000, 200, 200)

	// 2. GENERATE BG.TXT FILE FROM DOWNLOADED NEGATIVE (BACKGROUND) FILES

	files, err := ioutil.ReadDir("negatives")
	if err != nil {
		fmt.Println("err reading dir:", err)
	}
	var data string
	for _, file := range files {
		data += "negatives/" + file.Name() + "\n"
	}
	err = ioutil.WriteFile("bg.txt", []byte(data), 0666)
	if err != nil {
		log.Fatal(err)
	}

	// 3. CREATE POSITIVE SAMPLE VECTOR FILE
	// make training height and width variables
	// -w -h should be the width and height of your positive images
	createSampleCmdOptions := "-maxxangle 0.5 -maxyangle 0.5 maxzangle 0.5"
	samples.CreateSamples("adidas.png", "bg.txt", 1950, createSampleCmdOptions)

	// 5. TRAIN HAAR CASCADE FILE
	training.HaarCascade("data", 1800, 900, 20)
}