论文复现---BeatGAN

论文复现---BeatGAN

原创 小王搬运工 时序课堂 2024-06-11 10:02 四川

BeatGAN: Anomalous Rhythm Detection using Adversarially Generated Time Series

BeatGAN：使用对抗生成时间序列的异常心律检测

https://github.com/Vniex/BeatGAN

• PyTorch (1.0.0)

• scikit-learn (0.20.0)

• biosppy (0.6.1) # For data preprocess

• tqdm (4.28.1)

• matplotlib (3.0.2)

注意事项：

问题1：

这个地方按照这个安装我尝试了很多次都没安装成功，如果配置过国内镜像源如清华源，其中是没有老版本pytorch，清华源最早支持gpu的pytorch版本是1.5.0。而且torchvision的0.2.1这个版本即使是在pytorch自己的仓库都找不到，只能找到0.2.0，更别说在国内的镜像源中了。

解决方法：

去官网下载安装包进行安装：

https://download.pytorch.org/whl/cu100/torch_stable.html

然后找到torchvision，可以发现没有torch0.2.1，只有0.2.0，这里按照0.2.0安装也没有问题。

问题2：python版本不适配

问题3：GPU无法使用

这里我更改GPU检测部分，默认使用CPU运行一切正常！

其余的运行时提示缺少什么功能库，按照最新版默认安装即可。

https://www.dropbox.com/scl/fo/c14su187vh95gzchm7fgk/AGGykb6I5hMrbazgC8TDvPU?rlkey=rbzdfgp5y6bftpuwueqg7cpb1&e=1&dl=0

下载路径：BeatGAN\experiments\ecg\dataset\preprocessed\ano0

修改数据集加载路径：

本地电脑可以正常运行，由于性能原因未等到训练完，等待放到服务器上运行。

全套代码分享，包含数据集等，配好环境，没有问题即可运行。

链接：https://pan.baidu.com/s/1FUP-P6S4qm9WQZOPTJ3G-A

提取码：irb7

--来自百度网盘超级会员V6的分享