转换STM
使用Keras进行卷积LSTM进行视频分割。
数据应作为一组视频(mp4格式)在单独的文件夹中提供,并在文件名中带有后缀_label的相应分段掩码(扩展名之前)。
训练模型运行脚本
lstm_train_fcn.py
它将加载数据,将分辨率压缩4倍-输入的形状应分别为(W,D,C)(96,108,1)。 在这里,我只考虑1个通道,即黑白图像。 对于彩色图像,请更改形状。
在NVIDIA TitanX中,每1000帧视频序列的培训应该花费1个小时
使用卷积LSTM(中间)和仅卷积(顶部)进行超声视频序列和相应图像分割的示例
2021-09-22 16:14:51
196KB
Python
1