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

Compare commits

base: ch:f956b8b4d2260da9f33da825b1b8158c476dd563
Branches Tags
ch:master
...
compare: ch:f3ee7a4fb485ede950bc0579c252d9b2e0435c9f
Branches Tags
ch:master

2 Commits
f956b8b4d2 ... f3ee7a4fb4

Author SHA1 Message Date
Christoph Hagen
f3ee7a4fb4 Create folders for images 2023-01-17 15:09:03 +01:00
Christoph Hagen
026336f24d Don't attempt to fetch missing images in changed list 2023-01-17 15:08:44 +01:00
1 changed files with 4 additions and 0 deletions
Show Stats Expand all files Collapse all files

4
Training/train.swift
View File

@ -144,6 +144,7 @@ final class ClassifierCreator {
return nil return nil
} }
let changedImageList = await getChangedImageList() let changedImageList = await getChangedImageList()
.filter { $0.image < imageCounts[$0.cap] ?? 0 } // Filter non-existent images
if changedImageList.isEmpty { if changedImageList.isEmpty {
print("[INFO] No changed images to load") print("[INFO] No changed images to load")
@ -180,6 +181,9 @@ final class ClassifierCreator {
} }
private func loadImage(cap: Int, image: Int) async -> Bool { private func loadImage(cap: Int, image: Int) async -> Bool {
guard createFolderIfMissing(imageDirectory.appendingPathComponent(String(format: "%04d", cap))) else {
return false
}
let url = imageUrl(base: server.appendingPathComponent("images"), cap: cap, image: image) let url = imageUrl(base: server.appendingPathComponent("images"), cap: cap, image: image)
let tempFile: URL, response: URLResponse let tempFile: URL, response: URLResponse
do { do {