Update README.md
This commit is contained in:
parent
1386cbb3dc
commit
c9d4acd7a3
1 changed files with 46 additions and 0 deletions
46
README.md
46
README.md
|
@ -109,3 +109,49 @@ For installing torch under Windows see here: https://pytorch.org/get-started/loc
|
||||||
- heartbeat_scale_d = 1.0 / (heartbeat_scale_a + 1e-20)
|
- heartbeat_scale_d = 1.0 / (heartbeat_scale_a + 1e-20)
|
||||||
- result(x,y,t) = 1.0 + result_a(x,y,t) - result_d(x,y,t)
|
- result(x,y,t) = 1.0 + result_a(x,y,t) - result_d(x,y,t)
|
||||||
- end automatic_load
|
- end automatic_load
|
||||||
|
|
||||||
|
## DataContailer.py
|
||||||
|
|
||||||
|
### Constructor
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
path: str,
|
||||||
|
device: torch.device,
|
||||||
|
display_logging_messages: bool = False,
|
||||||
|
save_logging_messages: bool = False,
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
### automatic_load
|
||||||
|
def automatic_load(
|
||||||
|
self,
|
||||||
|
experiment_id: int = 1,
|
||||||
|
trial_id: int = 1,
|
||||||
|
start_position: int = 0,
|
||||||
|
start_position_coefficients: int = 100,
|
||||||
|
fs: float = 100.0,
|
||||||
|
use_regression: bool | None = False,
|
||||||
|
# Heartbeat
|
||||||
|
remove_heartbeat: bool = True, # i.e. use SVD
|
||||||
|
low_frequency: float = 5, # Hz Butter Bandpass Heartbeat
|
||||||
|
high_frequency: float = 15, # Hz Butter Bandpass Heartbeat
|
||||||
|
threshold: float | None = 0.5, # For the mask
|
||||||
|
# Extra exposed parameters:
|
||||||
|
align: bool = True,
|
||||||
|
iterations: int = 1, # SVD iterations: Do not touch! Keep at 1
|
||||||
|
lowrank_method: bool = True,
|
||||||
|
lowrank_q: int = 6,
|
||||||
|
remove_heartbeat_mean: bool = False,
|
||||||
|
remove_heartbeat_linear: bool = False,
|
||||||
|
bin_size: int = 4,
|
||||||
|
do_frame_shift: bool = True,
|
||||||
|
half_width_frequency_window: float = 3.0, # Hz (on side ) measure_heartbeat_frequency
|
||||||
|
mmap_mode: bool = True,
|
||||||
|
initital_mask_name: str | None = None,
|
||||||
|
initital_mask_update: bool = True,
|
||||||
|
initital_mask_roi: bool = False,
|
||||||
|
gaussian_blur_kernel_size: int | None = 3,
|
||||||
|
gaussian_blur_sigma: float = 1.0,
|
||||||
|
bin_size_post: int | None = None,
|
||||||
|
calculate_amplitude: bool = False,
|
||||||
|
) -> tuple[torch.Tensor, torch.Tensor | None]:
|
||||||
|
|
Loading…
Reference in a new issue