MaaAI icon indicating copy to clipboard operation
MaaAI copied to clipboard

優化與重構項目修改點(算是優化吧,請求技術交流)

Open crazysmile-PhD opened this issue 1 year ago • 0 comments

優化與重構項目修改點

  1. 參數化處理

  2. 代碼重用

  3. 錯誤處理與日誌

  4. 代碼結構整理

  5. 參數化處理 將硬編碼的參數(例如模型路徑、數據集路徑、訓練輪數等)轉換為命令行參數,允許在執行腳本時指定參數值。這樣可以使腳本更靈活,適應不同的環境和需求。 每個腳本(export.py、train.py、test.py)現在都可以通過命令行來傳遞參數,而不是將它們硬編碼在腳本內。

  6. 代碼重用 將重複使用的代碼片段(如模型加載、訓練、測試、模型導出等操作)提取到一個新的 common.py 模塊中。這樣做可以減少代碼重複,提高代碼的可維護性。 所有腳本現在都會從 common.py 中導入這些通用功能。

  7. 添加錯誤處理與日誌 在每個腳本中添加了基本的錯誤處理機制和日誌記錄功能。這樣在執行腳本時,如果出現錯誤,腳本能夠記錄錯誤信息,便於調試。 使用 Python 的 logging 模塊來記錄重要的操作步驟,讓你可以追踪腳本執行情況。

  8. 代碼結構整理 重新組織了代碼結構,確保所有腳本保持一致的風格和格式。 移除了一些潛在的冗餘代碼,簡化了腳本的邏輯流程。 具體文件修改 common.py: 新增了這個文件,包含了公共功能函數,如 load_model()、export_model()、train_model() 和 test_model()。 export.py、train.py、test.py: 這些文件經過重構,主要通過導入 common.py 中的功能來執行具體操作,並支持命令行參數。

crazysmile-PhD avatar Aug 09 '24 11:08 crazysmile-PhD