-
Notifications
You must be signed in to change notification settings - Fork 13
Description
角度范围-180到180不可以吗?为什么求正余弦要对角度的两倍求,直接原角度求正余弦再arctan2反求角度不行吗(前提角度限定范围在-180,180)?
还有刚试了试把标签当作网络输出去验证,不知道为什么都不匹配
`
import torch
import torch.optim as optim
import numpy as np
import matplotlib.pyplot as plt
import cv2
import time
#导入自定义包
from ggcnn import GGCNN
from cornell_pro import Cornell
from functions import post_process,detect_grasps,max_iou
from image_pro import Image
from grasp_pro import Grasp_cpaw
val_data = Cornell('图片标签路径',
random_rotate = True,
random_zoom = True,
output_size = 300)
val_dataset = torch.utils.data.DataLoader(val_data,batch_size = 1,shuffle = True)
for x,y,idx,rot,zoom in val_dataset:
q_out,ang_out,width_out = post_process(y[0], y[1], y[2], y[3])
grasps_pre = detect_grasps(q_out,ang_out,width_out,no_grasp = 1)
grasps_true = val_dataset.dataset.get_raw_grasps(idx,rot,zoom)
result = 0
for grasp_pre in grasps_pre:
if max_iou(grasp_pre,grasps_true) > 0.25:
result = 1
break
if result==0:
print(val_dataset.dataset.graspf[idx])
#break
`