ch/Caps-Train
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

specify thumbnail size

Browse Source
This commit is contained in:
Christoph Hagen 2023-11-19 23:32:28 +01:00
parent f67a9a2de7
commit 7e41818fd6
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Sources/ClassifierCreator.swift
View File

@ -9,6 +9,8 @@ final class ClassifierCreator {
private let sessionCheckpointIterations = 100 private let sessionCheckpointIterations = 100
private let sessionIterations = 1000 private let sessionIterations = 1000
private let thumbnailSize = 100
let server: URL let server: URL
let configuration: Configuration let configuration: Configuration
@ -456,7 +458,7 @@ final class ClassifierCreator {
let output = thumbnailDirectory.appendingPathComponent(String(format: "%04d.jpg", cap)) let output = thumbnailDirectory.appendingPathComponent(String(format: "%04d.jpg", cap))
do { do {
let command = "magick convert \(inputUrl.path) -quality 70% -resize 100x100 \(output.path)" let command = "magick convert \(inputUrl.path) -quality 70% -resize \(thumbnailSize)x\(thumbnailSize) \(output.path)"
let (code, output) = try safeShell(command) let (code, output) = try safeShell(command)
if code != 0 { if code != 0 {
throw TrainingError.failedToCreateThumbnail(cap, output) throw TrainingError.failedToCreateThumbnail(cap, output)