how to copy & modify nets model on tensorflow slim

想要修改tensorflow-slim 中 nets中的某个model,例如明明为kk_v2.py

观察到train_image_classifier.py中调用模型的部分

调用了nets_factory.get_network_fn,get_network如下:

 

我们看到model name 是通过 networks_map映射到func的

因此需要添加对于我们新的model,kk_v2的映射

由于train_image_classifier.py中有如下参数,

因此还需要修改预处理的映射表

还要修改arg_scopes_map,添加kk_v2的key

 

作者: CrazyKK

ex-ACMer@hust,stackoverflow-engineer@sensetime

说点什么

您将是第一位评论人!

提醒
wpDiscuz