Spaces:
Sleeping
Sleeping
| import os | |
| os.system("pip -qq install h5py") | |
| os.system("pip -qq install torch") | |
| os.system("pip -qq install torchvision") | |
| os.system("pip -qq install pandas") | |
| os.system("pip -qq install scikit-image") | |
| from dataset import FrameDataset | |
| import numpy as np | |
| import gradio as gr | |
| def generate_frames(ffile,dfile): | |
| dataset = FrameDataset(ffile=ffile,dfile=dfile) | |
| patches = dataset.get_peaks_skimage() | |
| normalized_patches = dataset.normalize_patches(patches) | |
| np.save("normalized_patches.npy",normalized_patches) | |
| return normalized_patches | |
| interface=gr.Interface(fn=generate_frames, inputs=[gr.File(label="frame file",type="binary"),gr.File(label="dark file",type="binary")], | |
| outputs=gr.outputs.File("peaks.npy")) | |
| interface.launch() | |