momodeltool 0.0.12

A small example package

moModelAnalysisTool

准备工作

pip install pqi
pqi use pypi 
pip install momodeltool / pip install momodeltool --upgrade

Visualization模块

通道可视化模块，使用举例：

from momodeltool import Visualization

'''使用开关'''
def forward(self, x):
  out = layer1(x)
  visual = Visualization()
  visual.switch_control()  # 打开展示开关，展示layer1
  visual.show_layer(out, module_name='layer1', show=True, save='./', pseudo=True, hist=True)
  visual.switch_control()  # 关闭展示开关，不展示layer2
  visual.show_layer(out, module_name='layer2', show=True, save='./', pseudo=True, hist=True) 
  out = layer2(out)
  visual.destroyAllWindows()  # 任意键入，关闭所有窗口

'''使用静态方法'''
def forward(self, x):
  out = layer1(x)
  Visualization.layer_show(out, module_name='layer1', show=True, save='./', pseudo=True, hist=True)  # 展示layer1，任意键入，关闭相关窗口
  out = layer2(out)

model.load_state_dict()
Visualization.kernel_show(model, module_name='test', show=Ture, save='./')

展示样例

ModelProcess模块

网络模型处理模块，使用样例：

from momodeltool import ModelProcess

mp = ModelProcess()
model = stereonet(batch_size=1, cost_volume_method=cost_volume_method)
pretrained_model_path = "mypath/model_1340.pth"
new_model = mp.auto_adapt(model, pretrained_model_path)  # 自动调整权重
mp.show_keys(new)  # 展示模型权重key