-
Notifications
You must be signed in to change notification settings - Fork 2.6k
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
[fix] fix image save path bug in Windows #1423
Conversation
Hi, thanks for your contribution. BTW, could this pr run correctly in Linux/Ubuntu OS? |
Pretty cool! So let it run and we can check out whether the number of train/val is identical. By the way, could you please fix the lint error? Thanks in advance! Best, |
The proceduce is pretty slow, after it finished, I will report the info. |
The generated train/val images numbers are 33978 and 11644, which is identical with iSAID |
|
* [fix] fix image save path bug in Windows * fix lint error
* [fix] fix image save path bug in Windows * fix lint error
* Flax: start adapting to Stable Diffusion 2 * More changes. * attention_head_dim can be a tuple. * Fix typos * Add simple SD 2 integration test. Slice values taken from my Ampere GPU. * Add simple UNet integration tests for Flax. Note that the expected values are taken from the PyTorch results. This ensures the Flax and PyTorch versions are not too far off. * Apply suggestions from code review Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Typos and style * Tests: verify jax is available. * Style * Make flake happy * Remove typo. * Simple Flax SD 2 pipeline tests. * Import order * Remove unused import. Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: @camenduru
Motivation
when run
.\tools\convert_datasets\isaid.py
in Windows OS, as the path str has no/
,there will get wrong image save path.Modification
using
os.path.basename(src_path).split('.')[0]
to get the filename without suffix.