图像旋转网络
给两张图片,判断旋转角度:
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):
la