andras/bipolaroidbipolaroid
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Small changes

Browse source
This commit is contained in:
Andras Schmelczer 2024-07-08 07:38:40 +01:00
parent 97e6fd8cd2
commit 51360171c4
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
3 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/histogram_transfer/__init__.py
View file

@ -1,4 +1,3 @@
from .regrain import regrain from .regrain import regrain
from .pdf_transfer_1d import pdf_transfer_1d
from .pdf_transfer_3d import pdf_transfer_3d from .pdf_transfer_3d import pdf_transfer_3d
from .apply_histogram import apply_histogram from .apply_histogram import apply_histogram

1
src/utils/__init__.py
View file

@ -5,3 +5,4 @@ from .kldiv import kldiv
from .set_up_logging import set_up_logging from .set_up_logging import set_up_logging
from .serialise_hparams import serialise_hparams from .serialise_hparams import serialise_hparams
from .get_device import get_device from .get_device import get_device
from .delete_corrupt_images import delete_corrupt_images

4
src/visualisation/plot_histograms_in_3d.py
View file

@ -6,7 +6,7 @@ import numpy as np
def plot_histograms_in_3d( def plot_histograms_in_3d(
histograms: Dict[str, np.ndarray], histograms_per_row: int = 3 histograms: Dict[str, np.ndarray], histograms_per_row: int = 3, height: int = 300
): ):
column_count = min(histograms_per_row, len(histograms)) column_count = min(histograms_per_row, len(histograms))
row_count = ceil(len(histograms) / histograms_per_row) row_count = ceil(len(histograms) / histograms_per_row)
@ -35,7 +35,7 @@ def plot_histograms_in_3d(
) )
for i in range(1, len(histograms) + 1) for i in range(1, len(histograms) + 1)
} }
fig.update_layout(**scenes, height=300 * column_count) fig.update_layout(**scenes, height=height * column_count)
fig.update_layout() # You can adjust the height as needed fig.update_layout() # You can adjust the height as needed
fig.show() fig.show()