def get_frame1(video_name): images = dataLoader_img(video_name,'007') #focal for img in images: frame = cv2.imread(img) yield frame
PATH_TO_DATA = os.path.join(CWD_PATH, modeldir) modelpth = glob.glob(PATH_TO_DATA + '/*')[0] #pretrained print(modelpth) #ROI를 지정한 첫번째 이미지를 불러옵니다. first_window = root + '/000/images/' + firstframe + '.png' anno = [] img = cv2.imread(first_window) cv2.namedWindow(first_window) cv2.imshow(first_window, img) img = cv2.putText(img, str(first_window.split('/')[-1]), (30, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0), 2) # setting ROI rect = cv2.selectROI(first_window, img, fromCenter=False, showCrosshair=True) anno.append( rect) # left, top, w, h = x, y, w, h 첫번째 꼭지점 (x,y) 두번째 꼭지점 (x+w,y+h) cv2.destroyWindow(first_window) if __name__ == '__main__': print('start') images = dataLoader_img( root, firstframe ) #root는 데이터 폴더 NonVideo4_tiny폴더이고 firstframe : lf 027번째 위치 focal show = None tracker = TrackerSiamFC(net_path=modelpth) tracker.track( img, images, show, anno[0], visualize=True ) #img는 첫 선택사진, images는 focal이미지, show는 보여줄 이미지, anno[0]은 첫 bbox
def get_plenframe(video_name, loc): images = dataLoader_img(video_name,loc) for img in images: frame = cv2.imread(img) yield frame
root = args.video_name lf = args.location CWD_PATH = os.getcwd() if modeldir: PATH_TO_DATA = os.path.join(CWD_PATH, modeldir) modelpth = glob.glob(PATH_TO_DATA + '/*')[0] #pretrained print(modelpth) first_window = root + '/000/images/' + lf + '.png' #007.png anno = [] img = cv2.imread(first_window) img = cv2.putText(img, str(first_window.split('/')[-1]), (30, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0), 2) cv2.namedWindow(first_window) cv2.imshow(first_window, img) # setting ROI rect = cv2.selectROI(first_window, img, fromCenter=False, showCrosshair=True) anno.append(rect) print(anno[0]) cv2.destroyWindow(first_window) if __name__ == '__main__': print('start') show = dataLoader_img(root, lf) focals = AllfocalLoader(root) tracker = TrackerSiamFC(net_path=modelpth) tracker.track(img, focals, show, anno[0], visualize=False)