|
|
@@ -55,7 +55,6 @@ func main() {
|
|
|
|
cli.SetDefaultValueFromEnv(&ociImage, "TRAIN_OCI_IMAGE", "")
|
|
|
|
cli.SetDefaultValueFromEnv(&ociImage, "TRAIN_OCI_IMAGE", "")
|
|
|
|
cli.SetDefaultValueFromEnv(&bucket, "TRAIN_BUCKET", "")
|
|
|
|
cli.SetDefaultValueFromEnv(&bucket, "TRAIN_BUCKET", "")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
flag.BoolVar(&debug, "debug", false, "Display debug logs")
|
|
|
|
flag.BoolVar(&debug, "debug", false, "Display debug logs")
|
|
|
|
|
|
|
|
|
|
|
|
displayFlags := flag.NewFlagSet("display", flag.ExitOnError)
|
|
|
|
displayFlags := flag.NewFlagSet("display", flag.ExitOnError)
|
|
|
@@ -87,13 +86,11 @@ func main() {
|
|
|
|
recordFlags.StringVar(&recordTopic, "mqtt-topic-records", os.Getenv("MQTT_TOPIC_RECORDS"), "Mqtt topic that contains record data for training, use MQTT_TOPIC_RECORDS if args not set")
|
|
|
|
recordFlags.StringVar(&recordTopic, "mqtt-topic-records", os.Getenv("MQTT_TOPIC_RECORDS"), "Mqtt topic that contains record data for training, use MQTT_TOPIC_RECORDS if args not set")
|
|
|
|
recordFlags.StringVar(&recordsPath, "record-path", os.Getenv("RECORD_PATH"), "Path where to write records files, use RECORD_PATH if args not set")
|
|
|
|
recordFlags.StringVar(&recordsPath, "record-path", os.Getenv("RECORD_PATH"), "Path where to write records files, use RECORD_PATH if args not set")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var basedir, destdir string
|
|
|
|
var basedir, destdir string
|
|
|
|
impdkFlags := flag.NewFlagSet("import-donkey-records", flag.ExitOnError)
|
|
|
|
impdkFlags := flag.NewFlagSet("import-donkey-records", flag.ExitOnError)
|
|
|
|
impdkFlags.StringVar(&basedir, "from", "", "source directory")
|
|
|
|
impdkFlags.StringVar(&basedir, "from", "", "source directory")
|
|
|
|
impdkFlags.StringVar(&destdir, "to", "", "destination directory")
|
|
|
|
impdkFlags.StringVar(&destdir, "to", "", "destination directory")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
trainingFlags := flag.NewFlagSet("training", flag.ExitOnError)
|
|
|
|
trainingFlags := flag.NewFlagSet("training", flag.ExitOnError)
|
|
|
|
trainingFlags.Usage = func() {
|
|
|
|
trainingFlags.Usage = func() {
|
|
|
|
fmt.Printf("Usage of %s %s:\n", os.Args[0], trainingFlags.Name())
|
|
|
|
fmt.Printf("Usage of %s %s:\n", os.Args[0], trainingFlags.Name())
|
|
|
@@ -133,7 +130,6 @@ func main() {
|
|
|
|
trainArchiveFlags.IntVar(&horizon, "horizon", 0, "Upper zone image to crop (in pixels)")
|
|
|
|
trainArchiveFlags.IntVar(&horizon, "horizon", 0, "Upper zone image to crop (in pixels)")
|
|
|
|
trainArchiveFlags.BoolVar(&withFlipImage, "with-flip-image", withFlipImage, "Flip horiontal image and reverse steering to increase data into training archive")
|
|
|
|
trainArchiveFlags.BoolVar(&withFlipImage, "with-flip-image", withFlipImage, "Flip horiontal image and reverse steering to increase data into training archive")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
modelsFlags := flag.NewFlagSet("models", flag.ExitOnError)
|
|
|
|
modelsFlags := flag.NewFlagSet("models", flag.ExitOnError)
|
|
|
|
modelsFlags.Usage = func() {
|
|
|
|
modelsFlags.Usage = func() {
|
|
|
|
fmt.Printf("Usage of %s %s:\n", os.Args[0], modelsFlags.Name())
|
|
|
|
fmt.Printf("Usage of %s %s:\n", os.Args[0], modelsFlags.Name())
|
|
|
@@ -249,7 +245,6 @@ func main() {
|
|
|
|
trainingFlags.PrintDefaults()
|
|
|
|
trainingFlags.PrintDefaults()
|
|
|
|
os.Exit(0)
|
|
|
|
os.Exit(0)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
case modelsFlags.Name():
|
|
|
|
case modelsFlags.Name():
|
|
|
|
|
|
|
|
|
|
|
|