Skip to content

Commit e29ddd9

Browse files
committed
Merge remote-tracking branch 'remotes/upstream/master' into refactor
# Updates: # cmd/deepsort/arguments.go # cmd/deepsort/classify.go # cmd/deepsort/main.go
2 parents a10f4d7 + 0b8a523 commit e29ddd9

File tree

1 file changed

+17
-6
lines changed

1 file changed

+17
-6
lines changed

arguments.go

Lines changed: 17 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,14 +9,17 @@ import (
99
)
1010

1111
var arguments = struct {
12-
Input string
13-
URL string
14-
DryRun bool
15-
Recursive bool
16-
Jobs int
17-
Network string
12+
Input string
13+
Output string
14+
OutputChoice bool
15+
URL string
16+
DryRun bool
17+
Recursive bool
18+
Jobs int
19+
Network string
1820
}{
1921
// Default arguments
22+
OutputChoice: false,
2023
Jobs: 1,
2124
}
2225

@@ -33,6 +36,12 @@ func argumentParsing(args []string) {
3336
Required: true,
3437
Help: "Your input folder."})
3538

39+
//output := parser.String("o", "output", &argparse.Options{
40+
// Required: false,
41+
// Help: "Your output folder, if output is set, " +
42+
// "original files will not be renamed, " +
43+
// "but the renamed version will be copied in the output folder."})
44+
3645
network := parser.Selector("n", "network", []string{"resnet-50", "googlenet"}, &argparse.Options{
3746
Required: false,
3847
Help: "The pre-trained deep neural network you want to use, " +
@@ -63,6 +72,8 @@ func argumentParsing(args []string) {
6372

6473
// Handle the input flag
6574
inputFolder, _ := filepath.Abs(*input)
75+
//arguments.Output = outputFolder
76+
//arguments.OutputChoice = true
6677
// Finally save the collected flags
6778
arguments.Network = *network
6879
arguments.Jobs = *jobs

0 commit comments

Comments
 (0)