我有估計(jì)姿勢(shì)的代碼。我正在嘗試實(shí)時(shí)運(yùn)行,但 OpenCV 沒(méi)有顯示視頻。如何解決這個(gè)錯(cuò)誤?我找不到任何問(wèn)題。當(dāng)我使用它時(shí)cv2.imshow("Video", dst)也不起作用。我的相機(jī)工作正常。我用小的 python 代碼試了一下。然后 imshow 也起作用了。當(dāng)我嘗試使用此代碼時(shí),它不起作用。我的相機(jī)是 USB 類(lèi)型的,我在 Ubuntu 平臺(tái)上工作。此代碼運(yùn)行時(shí)沒(méi)有任何錯(cuò)誤,唯一的問(wèn)題是 imshow 窗口未顯示。代碼:import jsonimport trt_pose.cocoimport trt_pose.modelsimport torchimport torch2trtfrom torch2trt import TRTModuleimport time, sysimport cv2import torchvision.transforms as transformsimport PIL.Imagefrom trt_pose.draw_objects import DrawObjectsfrom trt_pose.parse_objects import ParseObjectsimport argparseimport os.path'''hnum: 0 based human indexkpoint : keypoints (float type range : 0.0 ~ 1.0 ==> later multiply by image width, height'''def get_keypoint(humans, hnum, peaks): #check invalid human index kpoint = [] human = humans[0][hnum] C = human.shape[0] for j in range(C): k = int(human[j]) if k >= 0: peak = peaks[0][j][k] # peak[1]:width, peak[0]:height peak = (j, float(peak[0]), float(peak[1])) kpoint.append(peak) #print('index:%d : success [%5.3f, %5.3f]'%(j, peak[1], peak[2]) ) else: peak = (j, None, None) kpoint.append(peak) #print('index:%d : None %d'%(j, k) ) return kpointparser = argparse.ArgumentParser(description='TensorRT pose estimation run')parser.add_argument('--model', type=str, default='resnet', help = 'resnet or densenet' )args = parser.parse_args()with open('human_pose.json', 'r') as f: human_pose = json.load(f)topology = trt_pose.coco.coco_category_to_topology(human_pose)num_parts = len(human_pose['keypoints'])num_links = len(human_pose['skeleton'])if 'resnet' in args.model: print('------ model = resnet--------') MODEL_WEIGHTS = 'resnet18_baseline_att_224x224_A_epoch_249.pth' OPTIMIZED_MODEL = 'resnet18_baseline_att_224x224_A_epoch_249_trt.pth' model = trt_pose.models.resnet18_baseline_att(num_parts, 2 * num_links).cuda().eval() WIDTH = 224 HEIGHT = 224
1 回答

汪汪一只貓
TA貢獻(xiàn)1898條經(jīng)驗(yàn) 獲得超8個(gè)贊
你需要把
cv2.waitKey(1)
剛過(guò)
cv2.imshow("Video", execute(img, dst, t))
添加回答
舉報(bào)
0/150
提交
取消