feat/training_260202 #88

Merged
gina merged 3 commits from feat/training_260202 into develop 2026-02-13 10:51:07 +09:00
2 changed files with 15 additions and 1 deletions
Showing only changes of commit 12994aab60 - Show all commits

View File

@@ -206,7 +206,7 @@ public class TrainApiController {
}) })
@PostMapping("/counts/{uuid}") @PostMapping("/counts/{uuid}")
public ApiResponseDto<Void> getCount( public ApiResponseDto<Void> getCount(
@Parameter(description = "uuid", example = "80a0e544-36ed-4999-b705-97427f23337d") @Parameter(description = "uuid", example = "e22181eb-2ac4-4100-9941-d06efce25c49")
@PathVariable @PathVariable
UUID uuid) { UUID uuid) {
Long modelId = trainJobService.getModelIdByUuid(uuid); Long modelId = trainJobService.getModelIdByUuid(uuid);

View File

@@ -54,6 +54,20 @@ public class DataSetCountersService {
return (int) return (int)
stream.filter(Files::isRegularFile).filter(p -> p.toString().endsWith(".tif")).count(); stream.filter(Files::isRegularFile).filter(p -> p.toString().endsWith(".tif")).count();
} }
/*
대소문자 및 geojson 필요시
* try (var stream = Files.walk(dir)) {
return (int)
stream
.filter(Files::isRegularFile)
.filter(p -> {
String name = p.getFileName().toString().toLowerCase();
return name.endsWith(".tif") || name.endsWith(".geojson");
})
.count();
}
* */
} }
public DatasetCounters countTmpAfterBuild(Path path) throws IOException { public DatasetCounters countTmpAfterBuild(Path path) throws IOException {