rogerxavier
commited on
Commit
•
98e49c6
1
Parent(s):
53bee61
Update 0filterImage.py
Browse files- 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)
|