diff --git a/embed/radae_rx.py b/embed/radae_rx.py index 3751c0e..d13b82f 100644 --- a/embed/radae_rx.py +++ b/embed/radae_rx.py @@ -37,7 +37,7 @@ */ """ -import os, sys, struct +import os, sys, struct,argparse import numpy as np from matplotlib import pyplot as plt import torch @@ -257,6 +257,10 @@ def do_radae_rx(self, buffer_complex, features_out): return valid_output if __name__ == '__main__': + parser = argparse.ArgumentParser(description='RADAE streaming receiver, IQ.f32 on stdin to features.f32 on stdout') + parser.add_argument('model_name', type=str, help='path to model in .pth format', + default="../model19_check3/checkpoints/checkpoint_epoch_100.pth") + rx = radae_rx(model_name = "../model19_check3/checkpoints/checkpoint_epoch_100.pth") # allocate storage for output features diff --git a/src/rade_api.h b/src/rade_api.h index 9902eaf..7dd6694 100644 --- a/src/rade_api.h +++ b/src/rade_api.h @@ -70,7 +70,7 @@ int rade_n_tx_eoo_out(struct rade *r); int rade_nin_max(struct rade *r); int rade_n_features_in_out(struct rade *r); -// Note vocoder is not encapsulated in API in this version +// note vocoder is not encapsulated in API in this version // returns number of RADE_COMP samples written to tx_out[] int rade_tx(struct rade *r, RADE_COMP tx_out[], float features_in[]);