《一种左心室超声心动图视频的半监督异常检测方法》,哈工大学报,论文程序。请在根目录dataset下存放自己的数据,然后使用下述教程进行训练和测试:
一、运行环境要求:
- Ubuntu16.04
- Pytorch >= 0.4.1
二、模型训练:
- python -m visdom.server
- 训练:python train_*.py
三、模型测试:
- python -m visdom.server
- python validate_*.py
四、计算AUC结果: python evaluate/AUC_*.py
文件后缀Cardiac, CUHK分别代表对不同数据库进行训练的代码。读者可以下载开源数据库CUHK进行训练,也可以添加自己的数据进行训练。