diff --git a/detection/backdoor_detection.py b/detection/backdoor_detection.py index e50963c..e0c6398 100644 --- 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__":