修复bug

This commit is contained in:
leaf 2022-07-02 16:37:41 +08:00
parent 90c7612ea2
commit b7ee7599fb

2
gr.py
View File

@ -95,12 +95,12 @@ class Main:
def gr_img(self, filedir, diy): def gr_img(self, filedir, diy):
print(filedir) print(filedir)
img = cv2.imread(filedir)
if diy: if diy:
cnn = torch.load("CNN.pkl") cnn = torch.load("CNN.pkl")
cnn_two = torch.load("CNN_two.pkl") cnn_two = torch.load("CNN_two.pkl")
while True: while True:
not_match = 0 not_match = 0
img = cv2.imread(filedir)
img_tm = copy.deepcopy(img) img_tm = copy.deepcopy(img)
is_one_hand = self.at_main.gesture_recognition(self.at_detector, img, cnn_two) is_one_hand = self.at_main.gesture_recognition(self.at_detector, img, cnn_two)
if is_one_hand: if is_one_hand: