fix：逻辑小错误

2024-05-14 21:34:54 +08:00 · 2024-05-14 21:34:54 +08:00 · db3244f55a
commit db3244f55a
parent d073cfad31
1 changed files with 2 additions and 1 deletions
--- a/detection/backdoor_detection.py
+++ b/detection/backdoor_detection.py
@ -198,7 +198,8 @@ def main():
    # 如果未指定输出文件，则输出到 stdout；否则写入文件
    if args.pickle:
        pickleDataDetection(args.pickle, output_file)
-    process_path(args.path, output_format, args.mode, output_file)
+    else:
+        process_path(args.path, output_format, args.mode, output_file)


 if __name__ == "__main__":