rogerxavier commited on
Commit
98e49c6
1 Parent(s): 53bee61

Update 0filterImage.py

Browse files
Files changed (1) hide show
  1. 0filterImage.py +1 -1
0filterImage.py CHANGED
@@ -34,7 +34,7 @@ if __name__ == '__main__':
34
  if file.endswith(".jpg") or file.endswith(".png"):
35
  image_files.append(os.path.relpath(os.path.join(root, file)))
36
  for image_path in image_files:
37
- result = get_nsfw_score(image_path)#返回float的得分
38
  if result> 0.5:
39
  print("发现问题图片,需要删除以过审:",image_path)
40
  os.remove(image_path)
 
34
  if file.endswith(".jpg") or file.endswith(".png"):
35
  image_files.append(os.path.relpath(os.path.join(root, file)))
36
  for image_path in image_files:
37
+ result = get_nsfw_score(image_path,model)#返回float的得分
38
  if result> 0.5:
39
  print("发现问题图片,需要删除以过审:",image_path)
40
  os.remove(image_path)