robocars/robocar-steering-tflite-edgetpu
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: set model path

Browse Source
This commit is contained in:
Cyrille Nofficial 2021-10-12 11:44:47 +02:00
parent 0dfd7a595b
commit b7be75793b
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/rc-steering/rc-steering.go
View File

@ -53,7 +53,7 @@ func main() {
} }
defer client.Disconnect(50) defer client.Disconnect(50)
p := steering.NewPart(client, steeringTopic, cameraTopic, edgeVerbosity) p := steering.NewPart(client, modelPath, steeringTopic, cameraTopic, edgeVerbosity)
defer p.Stop() defer p.Stop()
cli.HandleExit(p) cli.HandleExit(p)

3
pkg/steering/steering.go
View File

@ -14,9 +14,10 @@ import (
"sort" "sort"
) )
func NewPart(client mqtt.Client, steeringTopic, cameraTopic string, edgeVerbosity int) *Part { func NewPart(client mqtt.Client, modelPath, steeringTopic, cameraTopic string, edgeVerbosity int) *Part {
return &Part{ return &Part{
client: client, client: client,
modelPath: modelPath,
steeringTopic: steeringTopic, steeringTopic: steeringTopic,
cameraTopic: cameraTopic, cameraTopic: cameraTopic,
edgeVebosity: edgeVerbosity, edgeVebosity: edgeVerbosity,