feat: 目录层级变更
This commit is contained in:
@@ -699,19 +699,13 @@ func (s *ScanService) runScan(ctx context.Context, rootDir, destDir string) erro
|
||||
nsfwDet := (*nsfw.Detector)(nil)
|
||||
s.setNsfwDetectorReady(false)
|
||||
if nsfwType == "human" {
|
||||
threshold := 0.5
|
||||
if v := s.store.GetConfig("nsfw.threshold"); v != "" {
|
||||
if t, err := strconv.ParseFloat(v, 64); err == nil && t > 0 && t < 1 {
|
||||
threshold = t
|
||||
}
|
||||
}
|
||||
workers := 4
|
||||
if v := s.store.GetConfig("nsfw.workers"); v != "" {
|
||||
if w, err := strconv.Atoi(v); err == nil && w > 0 {
|
||||
workers = w
|
||||
}
|
||||
}
|
||||
if d, err := nsfw.Start("", threshold, workers); err != nil {
|
||||
if d, err := nsfw.Start("", workers); err != nil {
|
||||
fmt.Printf("=== 色情识别未启用(图片照常处理): %v\n", err)
|
||||
} else {
|
||||
nsfwDet = d
|
||||
@@ -724,19 +718,13 @@ func (s *ScanService) runScan(ctx context.Context, rootDir, destDir string) erro
|
||||
// ---------- 第六步(可选):动漫色情识别 ----------
|
||||
animeNsfwDet := (*anime_nsfw.Detector)(nil)
|
||||
if nsfwType == "anime" {
|
||||
threshold := 0.5
|
||||
if v := s.store.GetConfig("anime.threshold"); v != "" {
|
||||
if t, err := strconv.ParseFloat(v, 64); err == nil && t > 0 && t < 1 {
|
||||
threshold = t
|
||||
}
|
||||
}
|
||||
workers := 2
|
||||
if v := s.store.GetConfig("anime.workers"); v != "" {
|
||||
if w, err := strconv.Atoi(v); err == nil && w > 0 {
|
||||
workers = w
|
||||
}
|
||||
}
|
||||
if d, err := anime_nsfw.Start("", threshold, workers); err != nil {
|
||||
if d, err := anime_nsfw.Start("", workers); err != nil {
|
||||
fmt.Printf("=== 动漫色情识别未启用(图片照常处理): %v\n", err)
|
||||
} else {
|
||||
animeNsfwDet = d
|
||||
|
||||
Reference in New Issue
Block a user