Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
crazysmile-PhD opened this issue Aug 9, 2024 · 0 comments

Comments

@crazysmile-PhD
Copy link

crazysmile-PhD commented Aug 9, 2024

優化與重構項目修改點

  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 中的功能來執行具體操作,並支持命令行參數。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant