diff --git a/tasks/item/slider.py b/tasks/item/slider.py index 9d4e8b950..af3c47ffe 100644 --- a/tasks/item/slider.py +++ b/tasks/item/slider.py @@ -49,7 +49,7 @@ class Slider: total = area[2] - area[0] parameters = { - 'prominence': 15, + 'prominence': 10, 'wlen': wlen, 'width': 2, 'distance': wlen / 2, @@ -58,6 +58,7 @@ class Slider: peaks, _ = signal.find_peaks(image, **parameters) peaks //= wlen diff = np.diff(peaks) + # print(diff) # Mask controller # Controller has orange border (240, 150, 57) and white center diff --git a/tasks/item/synthesize.py b/tasks/item/synthesize.py index d1683e8b5..c85cfcb64 100644 --- a/tasks/item/synthesize.py +++ b/tasks/item/synthesize.py @@ -37,7 +37,7 @@ def image_color_count(image, color, threshold=221): class WhiteStrip(Ocr): def pre_process(self, image): mask = color_similarity_2d(image, color=(255, 255, 255)) - mask = cv2.inRange(mask, 180, 255, dst=mask) + mask = cv2.inRange(mask, 160, 255, dst=mask) mask = np.mean(mask, axis=0) point = np.array(cv2.findNonZero(mask))[:, 0, 1] @@ -52,6 +52,13 @@ class SynthesizeItemName(OcrItemName, WhiteStrip): pass +class SynthesizeItemAmount(Digit, WhiteStrip): + def pre_process(self, image): + image = super().pre_process(image) + image = cv2.subtract((255, 255, 255, 0), image) + return image + + class SynthesizeInventoryManager(InventoryManager): @cached_property def dic_item_index(self): @@ -359,7 +366,7 @@ class Synthesize(CombatObtain, ItemUI): logger.hr('Synthesize amount set', level=2) slider = Slider(main=self, slider=SYNTHESIZE_SLIDER) slider.set(value, total) - ocr = Digit(SYNTHESIZE_AMOUNT, lang=server.lang) + ocr = SynthesizeItemAmount(SYNTHESIZE_AMOUNT, lang=server.lang) self.ui_ensure_index( value, letter=ocr, next_button=SYNTHESIZE_PLUS, prev_button=SYNTHESIZE_MINUS, interval=(0.1, 0.2))