Skip to content

Commit

Permalink
add model arg as input - but need major refactor to use aux data as i…
Browse files Browse the repository at this point in the history
…nput (sigh)
  • Loading branch information
drowe67 committed Sep 28, 2024
1 parent 9ac934d commit cb3518a
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
6 changes: 5 additions & 1 deletion embed/radae_rx.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion src/rade_api.h
Original file line number Diff line number Diff line change
Expand Up @@ -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[]);

Expand Down

0 comments on commit cb3518a

Please sign in to comment.