Skip to content

Latest commit

 

History

History
20 lines (14 loc) · 665 Bytes

README.md

File metadata and controls

20 lines (14 loc) · 665 Bytes

STAnoGAN

《一种左心室超声心动图视频的半监督异常检测方法》,哈工大学报,论文程序。请在根目录dataset下存放自己的数据,然后使用下述教程进行训练和测试:

一、运行环境要求:

  1. Ubuntu16.04
  2. Pytorch >= 0.4.1

二、模型训练:

  1. python -m visdom.server
  2. 训练:python train_*.py

三、模型测试:

  1. python -m visdom.server
  2. python validate_*.py

四、计算AUC结果: python evaluate/AUC_*.py

文件后缀Cardiac, CUHK分别代表对不同数据库进行训练的代码。读者可以下载开源数据库CUHK进行训练,也可以添加自己的数据进行训练。