作者好，有没有可以将裁剪之后的ROI输出出来的程序啊，试着写一直没写成

[SillyJumbo]

No description provided.

[dxbdxx]

bbox_pred scale后可以定位到正方形4个点，求出正方形中心c, 然后按照theta_pred在c上进行中心旋转，最后就可以按照行列截取。思路是这样的，以上操作在cv2里都有，代码我已经没了

[SillyJumbo]

程序中的img的维度显示为16 2 3 768 1024，我输出出来的ROI1和ROI2图片全是黑色的，这个img的维度咋弄啊，中间那个2通道一直去不掉，可以给点建议吗

[dxbdxx]

维度分别是 batch_size rgb/ir channels height width. 取一张图片那就是 img[0,0] 另外这个代码需要训练，我这没有提供训练好的模型。不训练的话输出都是全黑的。