triplet-loss minist
程序员文章站
2024-03-14 21:17:41
...
Code:https://github.com/SpikeKing/triplet-loss-mnist
记录一下在跑这个代码时遇到的一些问题以及解决办法:
Environment:
python2.7
keras ‘2.2.4’
tensorflow ‘1.13.1’
numpy ‘1.16.4’
pydot ‘1.2.4’
graphviz ‘2.40.1’
bunch ‘1.0.1’
问题一:
OSError: `pydot` failed to call GraphViz.Please install GraphViz (https://www.graphviz.org/) and ensure that its executables are in the $PATH.
解决办法:
pip install pydot-ng
pip install pydot==1.2.4
conda install graphviz
问题二:
TypeError: Required Group, str or dict. Received: <type 'unicode'>.
解决办法:
triplet_trainer.py 87行
self.model.save(str(os.path.join(self.config.cp_dir,"triplet_loss_model.h5")))
triplet_infer.py 36行
model_path = str(os.path.join(self.config.cp_dir, "triplet_loss_model.h5"))
运行python main_train.py -c configs/triplet_config.json
训练截图:
运行python main_test.py -c configs/triplet_config.json
测试截图: