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

New features & bug fixes #5

Merged
merged 49 commits into from
Sep 15, 2020
Merged

New features & bug fixes #5

merged 49 commits into from
Sep 15, 2020

Conversation

ProGamerGov
Copy link
Owner

@ProGamerGov ProGamerGov commented Sep 11, 2020

Bug Fixes:

  • Resolved confusion between BGR and RGB usage. Some custom models may need to be retrained.
  • Visualization scripts can now load all 3 starter models without errors.
  • The -seed parameter in the train_googlenet.py script should work more effectively now. Though this may not be the case if multiple training or validation workers are used.

Changes:

  • The train_googlenet.py script now saves mean and standard deviation values in BGR format rather than RGB format.
  • The visualization scripts now expects mean and standard deviation values in BGR format rather than RGB format.
  • The calc_ms.py script now outputs normalization values in BGR format rather than RGB.
  • The vis_fc.py script has been replaced with vis_multi.py.

New Features:

  • The new vis_multi.py script lets you visualize all channels in any specified layer, and lets you select a visualization batch size among other new features.
  • Added new tool to edit model file values, called edit_model.py.
  • Added new normalization value format attribute to models for easier handling of BGR and RGB models.
  • Added -use_rgb parameter to the calc_ms.py script for if you want the original behavior.

To update your old models to the new correct format:

python data_tools/edit_model.py -model_file <your-model.pth> -normval_format bgr -reverse_normvals -output_name <updated-model.pth>

@ProGamerGov ProGamerGov merged commit b912f21 into master Sep 15, 2020
@ProGamerGov ProGamerGov deleted the Update branch September 15, 2020 16:40
@ProGamerGov
Copy link
Owner Author

I also changed neural-dream's behavior for loading Dream-Creator models: ProGamerGov/neural-dream@e8cd4e6

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.

None yet

1 participant