Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/WheatCao/ICCV2025-IRGPT
/ functions
Functions
13 in github.com/WheatCao/ICCV2025-IRGPT
⨍
Functions
13
◇
Types & classes
0
↓ 2 callers
Function
extract_boxes
(s: str)
test_utils/Detection.py:7
↓ 2 callers
Function
extract_number
从字符串中提取 <num> 标签内的数字 :param s: 输入字符串 :return: 提取的数字(整数),如果未找到则返回 None
test_utils/Counting.py:15
↓ 2 callers
Function
extract_points
从字符串中提取<pos>标签内的点列表
test_utils/Location.py:11
↓ 1 callers
Function
calculate_iou
计算两个边界框的 IoU
test_utils/Detection.py:46
↓ 1 callers
Function
calculate_metrics
计算人群计数任务的评估指标(MAE, MSE, RMSE) :param data: JSON 数据(字典列表) :return: MAE, MSE, RMSE
test_utils/Counting.py:26
↓ 1 callers
Function
calculate_sample_accuracy
计算单个样本的检测精度
test_utils/Detection.py:60
↓ 1 callers
Function
distance
计算两个点之间的欧氏距离
test_utils/Location.py:7
↓ 1 callers
Function
evaluate_json
评估JSON文件中模型预测结果的有效性。 输入: json_file: JSON文件路径 threshold: 匹配阈值,默认5.0 输出: 包含精确率、召回率、F1分数、全局平均距离和样本平均MAE的字典
test_utils/Location.py:59
↓ 1 callers
Function
load_json
读取 JSON 文件 :param file_path: JSON 文件路径 :return: JSON 数据(字典列表)
test_utils/Counting.py:5
↓ 1 callers
Function
match_points
匹配真实点和预测点,确保每个真实点匹配到最近的预测点,且每个预测点只能匹配一次。 返回匹配对、未匹配的真实点和未匹配的预测点。
test_utils/Location.py:29
Function
calculate_accuracy
(json_file)
test_utils/Recognition.py:3
Function
evaluate_crowd_counting
评估人群计数任务的模型性能 :param file_path: JSON 文件路径 :return: MAE, MSE, RMSE
test_utils/Counting.py:49
Function
evaluate_detection_accuracy
计算整个 JSON 文件中所有样本的平均检测精度
test_utils/Detection.py:76