Awesome
This is the part of the code about keypoint matching.
Unfortunately we lost the original dumped dataset, and only keep the way to generate the dataset. And for the reason that the old code is hard to trace, we reproduce the code according to the paper in the new version of tensorflow (1.12.0) recently. So the result inevitably be different with the result reported in the paper. Some of them are higher, some of them are lower. But very close to the paper.
usage
--data_tr
, --data_va
and --data_te
control the dataset for training, validation and testing.
We trained the model on the dataset brown_bm_3_05
, if you want to test on the dataset herzjesu
, just run the following line
python main.py --run_mode="test" --data_te="herzjesu"
dumping dataset uses
python dump_data.py --data_tr=`brown_bm_3_05` --data_te=`brown_bm_3_05` --data_te=`brown_bm_3_05`
using my trained model to test the dataset herzjesu
python main.py --log_dir="mytrain" --data_te=`herzjesu`
If you encounter any path problem, I believe you guys could fix them.