register_net

CNN for registering two images- calculates y shift only as of now.

Download data

wget --no-check-certificate https://storage.googleapis.com/mledu-datasets/cats_and_dogs_filtered.zip -O cats_and_dogs_filtered.zip

unzip cats_and_dogs_filtered -d cat_dog

python train.py

python infer.py

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
samples		samples
weights		weights
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
config.py		config.py
datagen.py		datagen.py
infer.py		infer.py
reg_cnn.py		reg_cnn.py
train.py		train.py