今天的内容呢,是制作数据集,非常重要。包括上一篇文章的转为标签形式,还有这次的旋转矫正图像,初步裁剪图像划分训练集和测试集,以及裁剪为固定大小的图像形成数据集。
旋转矫正代码
interactive_rotate_with_pt.py
具体操作如下:运行代码后,图中会出现一个机器默认的矫正方案,如果想手动选取点,则点右键取消。用左键点击想要矫正的四个点,形成绿色的框,但不是矩形。然后会形成红色的矩形框,为矫正后的预览图。觉得合适的话可单击滚轮,图像将会像红色框一样旋转,并保存图像。
但是我的图像还算比较正,矫正的意义也不大,故没有矫正啦~
初步裁剪
这一步需要手动划分训练集和测试集,并初步裁剪感兴趣区域。
由于只有三张拼接的图像,所以要慎重选择训练集和测试集,保证三种农作物尽量均匀地分布在训练集和测试集中。其中kivi fruit只有一块地,所以要对半裁开。而rape每块地的差别有点大,也需要一分为二。
具体的操作: