[Test] Fix release task problem (#520)

* Fix release task

* Update compare_with_gt.py

* fix code style
This commit is contained in:
Jason
2022-11-08 20:12:22 +08:00
committed by GitHub
parent b0af754cfa
commit c7277ef4ac

View File

@@ -3,31 +3,31 @@ import re
diff_score_threshold = { diff_score_threshold = {
"linux-x64": { "linux-x64": {
"label_diff": 0, "label_diff": 1e-02,
"score_diff": 1e-4, "score_diff": 1e-4,
"boxes_diff_ratio": 1e-4, "boxes_diff_ratio": 1e-4,
"boxes_diff": 1e-3 "boxes_diff": 1e-3
}, },
"linux-aarch64": { "linux-aarch64": {
"label_diff": 0, "label_diff": 1e-02,
"score_diff": 1e-4, "score_diff": 1e-4,
"boxes_diff_ratio": 1e-4, "boxes_diff_ratio": 1e-4,
"boxes_diff": 1e-3 "boxes_diff": 1e-3
}, },
"osx-x86_64": { "osx-x86_64": {
"label_diff": 0, "label_diff": 1e-02,
"score_diff": 1e-4, "score_diff": 1e-4,
"boxes_diff_ratio": 2e-4, "boxes_diff_ratio": 2e-4,
"boxes_diff": 1e-3 "boxes_diff": 1e-3
}, },
"osx-arm64": { "osx-arm64": {
"label_diff": 0, "label_diff": 1e-02,
"score_diff": 1e-4, "score_diff": 1e-4,
"boxes_diff_ratio": 2e-4, "boxes_diff_ratio": 2e-4,
"boxes_diff": 1e-3 "boxes_diff": 1e-3
}, },
"win-x64": { "win-x64": {
"label_diff": 0, "label_diff": 1e-02,
"score_diff": 5e-4, "score_diff": 5e-4,
"boxes_diff_ratio": 1e-3, "boxes_diff_ratio": 1e-3,
"boxes_diff": 1e-3 "boxes_diff": 1e-3
@@ -35,6 +35,14 @@ diff_score_threshold = {
} }
def all_sort(x):
x1 = x.T
y = np.split(x1, len(x1))
z = list(reversed(y))
index = np.lexsort(z)
return x[index]
def parse_arguments(): def parse_arguments():
import argparse import argparse
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
@@ -83,8 +91,8 @@ def write2file(error_file):
with open(error_file, "a+") as f: with open(error_file, "a+") as f:
from platform import python_version from platform import python_version
py_version = python_version() py_version = python_version()
f.write(args.platform + " " + py_version + " " + f.write(args.platform + " " + py_version + " " + args.result_path.split(
args.result_path.split(".")[0] + "\n") ".")[0] + "\n")
def save_numpy_result(file_path, error_msg): def save_numpy_result(file_path, error_msg):
@@ -134,4 +142,6 @@ if __name__ == '__main__':
gt_numpy = convert2numpy(args.gt_path, args.conf_threshold) gt_numpy = convert2numpy(args.gt_path, args.conf_threshold)
infer_numpy = convert2numpy(args.result_path, args.conf_threshold) infer_numpy = convert2numpy(args.result_path, args.conf_threshold)
gt_numpy = all_sort(gt_numpy)
infer_numpy = all_sort(infer_numpy)
check_result(gt_numpy, infer_numpy, args) check_result(gt_numpy, infer_numpy, args)