2024-06-03 20:31:35 +08:00
1 changed files with 2 additions and 1 deletions
--- a/detection/backdoor_detection.py
+++ b/detection/backdoor_detection.py
@ -198,6 +198,7 @@ def main():
    # 如果未指定输出文件，则输出到 stdout；否则写入文件
    if args.pickle:
        pickleDataDetection(args.pickle, output_file)
+    else:
        process_path(args.path, output_format, args.mode, output_file)