mirror of
https://github.com/EyeTrackVR/EyeTrackVR.git
synced 2025-11-04 14:39:42 +08:00
Fixes import to work correctly in non-PyCharm environments
This commit is contained in:
parent
7e1c138136
commit
6019f44177
@ -1,3 +1,4 @@
|
|||||||
|
import sys
|
||||||
import math
|
import math
|
||||||
import os
|
import os
|
||||||
import timeit
|
import timeit
|
||||||
@ -8,9 +9,15 @@ import cv2
|
|||||||
import numpy as np
|
import numpy as np
|
||||||
from numpy.linalg import _umath_linalg
|
from numpy.linalg import _umath_linalg
|
||||||
|
|
||||||
from EyeTrackApp.utils.img_utils import safe_crop
|
if os.environ.get("PyCharm", None) is None:
|
||||||
from EyeTrackApp.utils.misc_utils import clamp
|
sys.path.append("../")
|
||||||
from EyeTrackApp.utils.time_utils import FPSResult, TimeitResult, format_time
|
from utils.img_utils import safe_crop # noqa
|
||||||
|
from utils.misc_utils import clamp # noqa
|
||||||
|
from utils.time_utils import FPSResult, TimeitResult, format_time # noqa
|
||||||
|
else:
|
||||||
|
from EyeTrackApp.utils.img_utils import safe_crop
|
||||||
|
from EyeTrackApp.utils.misc_utils import clamp
|
||||||
|
from EyeTrackApp.utils.time_utils import FPSResult, TimeitResult, format_time
|
||||||
|
|
||||||
# from line_profiler_pycharm import profile
|
# from line_profiler_pycharm import profile
|
||||||
|
|
||||||
@ -29,7 +36,7 @@ loop_num = 1 if imshow_enable or save_img or save_video else 100
|
|||||||
input_video_path = "Pro_demo2.mp4"
|
input_video_path = "Pro_demo2.mp4"
|
||||||
output_img_path = f'./{this_file_name}_{alg_ver}_new.png' if not old_mode else f'./{this_file_name}_{alg_ver}_old.png'
|
output_img_path = f'./{this_file_name}_{alg_ver}_new.png' if not old_mode else f'./{this_file_name}_{alg_ver}_old.png'
|
||||||
output_video_path = f'./{this_file_name}_{alg_ver}_new.mp4' if not old_mode else f'./{this_file_name}_{alg_ver}_old.mp4'
|
output_video_path = f'./{this_file_name}_{alg_ver}_new.mp4' if not old_mode else f'./{this_file_name}_{alg_ver}_old.mp4'
|
||||||
logfilename = f'./{this_file_name}_{alg_ver}_new.log' if not old_mode else f'./{this_file_name}_old.log'
|
logfilename = f'./{this_file_name}_{alg_ver}_new.log' if not old_mode else f'./{this_file_name}_{alg_ver}_old.log'
|
||||||
print_enable = False # I don't recommend changing to True.
|
print_enable = False # I don't recommend changing to True.
|
||||||
|
|
||||||
# RANSAC
|
# RANSAC
|
||||||
@ -881,7 +888,7 @@ class HSRAC_cls(object):
|
|||||||
|
|
||||||
self.cap = None
|
self.cap = None
|
||||||
|
|
||||||
self.timedict = {"to_gray": [], "int_img": [], "conv_int": [], "crop": [], "ransac": [], "total_cv": []}
|
self.timedict = {"to_gray": [], "int_img": [], "hsf": [], "crop": [], "ransac": [], "total_cv": []}
|
||||||
|
|
||||||
# ransac
|
# ransac
|
||||||
self.rng = np.random.default_rng()
|
self.rng = np.random.default_rng()
|
||||||
@ -976,7 +983,7 @@ class HSRAC_cls(object):
|
|||||||
# Pseudo-visualization of HSF
|
# Pseudo-visualization of HSF
|
||||||
# cv2.normalize(cv2.filter2D(cv2.filter2D(frame_pad, cv2.CV_64F, hsf.get_kernel()[hsf.get_kernel().shape[0]//2,:].reshape(1,-1), borderType=cv2.BORDER_CONSTANT), cv2.CV_64F, hsf.get_kernel()[:,hsf.get_kernel().shape[1]//2].reshape(-1,1), borderType=cv2.BORDER_CONSTANT),None,0,255,cv2.NORM_MINMAX,dtype=cv2.CV_8U))
|
# cv2.normalize(cv2.filter2D(cv2.filter2D(frame_pad, cv2.CV_64F, hsf.get_kernel()[hsf.get_kernel().shape[0]//2,:].reshape(1,-1), borderType=cv2.BORDER_CONSTANT), cv2.CV_64F, hsf.get_kernel()[:,hsf.get_kernel().shape[1]//2].reshape(-1,1), borderType=cv2.BORDER_CONSTANT),None,0,255,cv2.NORM_MINMAX,dtype=cv2.CV_8U))
|
||||||
|
|
||||||
self.timedict["conv_int"].append(timeit.default_timer() - conv_int_start_time)
|
self.timedict["hsf"].append(timeit.default_timer() - conv_int_start_time)
|
||||||
|
|
||||||
crop_start_time = timeit.default_timer()
|
crop_start_time = timeit.default_timer()
|
||||||
# Define the center point and radius
|
# Define the center point and radius
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user