50 lines
1.7 KiB
Swift
50 lines
1.7 KiB
Swift
import ArgumentParser
|
|
import Foundation
|
|
|
|
@main
|
|
struct CapTrain: AsyncParsableCommand {
|
|
|
|
@Argument(help: "The path to the configuration file")
|
|
var configPath: String?
|
|
|
|
@Option(name: .shortAndLong, help: "The number of iterations to train")
|
|
var iterations: Int?
|
|
|
|
@Option(name: .shortAndLong, help: "The url of the caps server")
|
|
var serverPath: String?
|
|
|
|
@Option(name: .shortAndLong, help: "The authentication token for the server")
|
|
var authentication: String?
|
|
|
|
@Option(name: .shortAndLong, help: "The folder where the content (images, classifier, thumbnails) is stored")
|
|
var folder: String?
|
|
|
|
func run() async throws {
|
|
let configurationFile = try configurationFile()
|
|
guard let contentFolder = folder ?? configurationFile?.contentFolder,
|
|
let trainingIterations = iterations ?? configurationFile?.trainingIterations,
|
|
let serverPath = serverPath ?? configurationFile?.serverPath,
|
|
let authenticationToken = authentication ?? configurationFile?.authenticationToken
|
|
else {
|
|
throw TrainingError.missingArguments
|
|
}
|
|
|
|
let configuration = Configuration(
|
|
contentFolder: contentFolder,
|
|
trainingIterations: trainingIterations,
|
|
serverPath: serverPath,
|
|
authenticationToken: authenticationToken)
|
|
|
|
let creator = try ClassifierCreator(configuration: configuration)
|
|
try await creator.run()
|
|
}
|
|
|
|
private func configurationFile() throws -> ConfigurationFile? {
|
|
guard let configPath else {
|
|
return nil
|
|
}
|
|
let configurationFileUrl = URL(fileURLWithPath: configPath)
|
|
return try ConfigurationFile(at: configurationFileUrl)
|
|
}
|
|
}
|