cnn旋转角度判断回归

图像旋转网络

给两张图片，判断旋转角度：

https://github.com/jidai-code/ExplicitOrientationEstimator

自己复现了一下，并尝试改进：

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.nn import init


def conv2d(in_channels, out_channels, kernel_size, stride, padding):
	return nn.Sequential(
		nn.Conv2d(in_channels=in_channels, out_channels=out_channels, kernel_size=kernel_size, stride=stride,
				  padding=padding, bias=False), nn.BatchNorm2d(out_channels), nn.ReLU(inplace=True), )


def make_layers(cfg, in_channels):
	layers = []
	for l in cfg:
		if l == 'M':
			layers += [nn.MaxPool2d(kernel_size=2, stride=2)]
		else:
			layers += [conv2d(in_channels=in_channels, out_c