Skip to content

Conversation

k2-gc
Copy link

@k2-gc k2-gc commented May 5, 2025

Thanks for your great work on this repository!

In this PR, I applied the following modifications to support Python 3.12 and updated dependencies:

  • requirements.txt
    Added required libraries with version specifications.

  • train.py / val.py

    1. Updated import statements and usage to be compatible with albumentations==2.0.6.
    2. Added logic to use GPU only if available.
  • dataset.py
    If self.transform includes Normalize(), skip dividing image pixels by 255.0.


NOTE:
This modification is not backward compatible and may cause errors when used with older library versions.

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

Successfully merging this pull request may close these issues.

1 participant