复制图片名的图片查看器

由于最近要做一个图片数据集标注，需要对图片进行定向截图，并且保证截图名与原图片名一致，就做了这个小脚本，配合QQ截图使用。

代码

import win32clipboard as w
import win32con
import cv2
import keyboard
from pathlib import Path

flag = True
over = False
logPath = 'D:\\dataset\\capture.txt'#起始图片日志保存目录

def set_text(str):
    """ 将str写入剪贴板 """
    w.OpenClipboard()
    w.EmptyClipboard()
    w.SetClipboardData(win32con.CF_UNICODETEXT, str)
    w.CloseClipboard()


def down():
    global flag
    flag = True


def up():
    global flag
    flag = False


def end():
    global over
    over = True

def origin():
    my_file = Path(logPath)
    if my_file.exists():
        f = open(logPath, 'r')
        i = int(f.read())
        f.close()
    else:
        f = open(logPath, 'w')
        f.write('0')
        f.close()
        i = 0
    return i

if __name__ == '__main__':
    i = origin()

    while (1):
        fileDir = 'D:\dataset\luo_01\\%06d.jpg' % (i)#图片目录
        str = '%06d.jpg' % (i)
        set_text(str)
        keyboard.add_hotkey('D', down)#D下一张图
        keyboard.add_hotkey('A', up)#A上一张图
        keyboard.add_hotkey('Q', end)#Q关闭窗口并将当前图片序号保存至日志文件
        src = cv2.imread(fileDir)
        src = cv2.resize(src, (1366, 768))
        cv2.imshow('%d' % (i), src)
        cv2.resizeWindow('%d' % (i), 1366, 768);
        cv2.moveWindow('%d' % (i), 0, 0)
        cv2.waitKey(0)
        cv2.destroyAllWindows()
        if (over):
            f = open(logPath, 'w')
            f.write('%d' % i)
            f.close()
            exit(0)
        if (flag):
            i += 1
        else:
            i -= 1

第一次运行时会生成日志文件
下次程序运行时就从这个数据开始打开图片，按D下一张图，按A上一张图。
打开图片时会直接将文件名写入剪贴板，用QQ截图ctrl+alt+a后ctrl+s，再ctrl+v直接粘贴回车即可。