支持文件夹扫描吗

[qq14909]

作者，你好，想请教下
能支持文件夹扫描吗，我测试后发现只能一张图片运行

(myenv) E:\watermark-tracer>python trace.py --no_api -m E:\watermark-tracer\img\084023n0yummfazg7m8apt.jpg
YOLOv5 2024-2-27 Python-3.8.18 torch-2.2.1+cpu CPU

Fusing layers...
Model summary: 367 layers, 46113663 parameters, 0 gradients
Adding AutoShape...
YoloV5加载成功
检测中
Yolo检测失败

另外有考虑增加识别水印位置调用其它软件去除吗，我想调用这个https://github.com/Sanster/IOPaint?tab=readme-ov-file自动去除

[Kamino666]

新的功能可以fork之后添加哦，这个是两年前我一门课的大作业，所以不会有更新啦~