感谢
首先先感谢这位老哥的无私分享,老哥牛!我直接使用了人家的模型,正确率超高!!!
奉上大哥的github:觉得有用的同学可以支持一下
RotNet
This repository contains the code necessary to train and test convolutional neural networks (CNNs) for predicting the rotation angle of an image to correct its orientation. There are scripts to train two models, one on and another one on the . Since the data for this application is generated on-the-fly, you can also train