czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

将检测数据集逻辑挪到顶部

This commit is contained in:
czzhangheng 2025-11-11 18:59:12 +08:00
parent 2ba061e57a
commit 8e53d25ab1
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
run.py
View File

@ -1,5 +1,7 @@
import torch import torch
from utils.Download_data import check_and_download_data
data_complete = check_and_download_data()
assert data_complete is not None, "数据集下载失败，请重试！"
# import time # import time
from config.args_parser import parse_args from config.args_parser import parse_args
@ -58,10 +60,5 @@ def main():
case _: case _:
raise ValueError(f"Unsupported mode: {args['basic']['mode']}") raise ValueError(f"Unsupported mode: {args['basic']['mode']}")
if __name__ == "__main__": if __name__ == "__main__":
from utils.Download_data import check_and_download_data
data_complete = check_and_download_data()
assert data_complete is not None, "数据集下载失败，请重试！"
main() main()