sangge/BackDoorBuster
sangge/BackDoorBuster
1
0
Fork 0
Code Issues Pull Requests Projects 1 Activity

perf: 修改检测结果输出路径；删除测试模块

Browse Source
This commit is contained in:
dqy 2024-04-22 17:07:50 +08:00
parent 278e9ee42e
commit f2d4e1befc
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
detection/backdoor_detection.py
View File

@ -1,3 +1,5 @@
# Usage: python backdoor_detection.py <code_path> <output_format>
import os import os
import re import re
import sys import sys
@ -70,7 +72,7 @@ def output_results(
results: Dict[str, List[Tuple[int, str]]], output_format: str, file_path: str results: Dict[str, List[Tuple[int, str]]], output_format: str, file_path: str
): ):
# Create the 'results' directory if it does not exist # Create the 'results' directory if it does not exist
results_dir = "../results" results_dir = "../results/code"
if not os.path.exists(results_dir): if not os.path.exists(results_dir):
os.makedirs(results_dir) os.makedirs(results_dir)
@ -146,10 +148,6 @@ def process_path(path: str, output_format: str):
sys.exit(1) sys.exit(1)
def test():
print("hello world")
def main(): def main():
if len(sys.argv) < 3: if len(sys.argv) < 3:
print("Usage: python backdoor_detection.py <path> <output_format>") print("Usage: python backdoor_detection.py <path> <output_format>")