優化與重構項目修改點(算是優化吧,請求技術交流)
crazysmile-PhD opened this issue · 0 comments
crazysmile-PhD commented
優化與重構項目修改點
-
參數化處理
-
代碼重用
-
錯誤處理與日誌
-
代碼結構整理
-
參數化處理
將硬編碼的參數(例如模型路徑、數據集路徑、訓練輪數等)轉換為命令行參數,允許在執行腳本時指定參數值。這樣可以使腳本更靈活,適應不同的環境和需求。
每個腳本(export.py、train.py、test.py)現在都可以通過命令行來傳遞參數,而不是將它們硬編碼在腳本內。 -
代碼重用
將重複使用的代碼片段(如模型加載、訓練、測試、模型導出等操作)提取到一個新的 common.py 模塊中。這樣做可以減少代碼重複,提高代碼的可維護性。
所有腳本現在都會從 common.py 中導入這些通用功能。 -
添加錯誤處理與日誌
在每個腳本中添加了基本的錯誤處理機制和日誌記錄功能。這樣在執行腳本時,如果出現錯誤,腳本能夠記錄錯誤信息,便於調試。
使用 Python 的 logging 模塊來記錄重要的操作步驟,讓你可以追踪腳本執行情況。 -
代碼結構整理
重新組織了代碼結構,確保所有腳本保持一致的風格和格式。
移除了一些潛在的冗餘代碼,簡化了腳本的邏輯流程。
具體文件修改
common.py: 新增了這個文件,包含了公共功能函數,如 load_model()、export_model()、train_model() 和 test_model()。
export.py、train.py、test.py: 這些文件經過重構,主要通過導入 common.py 中的功能來執行具體操作,並支持命令行參數。