심플 어노테이션 툴 - 첫번째

import glob # 파일들을 잡기 위한 패키지
import cv2 # 이미지 파일을 처리하기 위한 패키지

img_files = glob.glob('./data/*.jpg') # data폴더 안에 있는 모든 jpg 파일들 목록을 가져옵니다.

cv2.namedWindow('tool') # tool이란 윈도우(창)을 하나 띄웁니다.

for img_file in img_files: # 모든 jpg 파일들 목록에서 jpg 파일명을 하나하나씩 가지고 옵니다.
    img = cv2.imread(img_file) # 가지고 온 jpg 파일명으로부터 이미지를 불러옵니다.
    cv2.imshow('tool', img) # 불러온 이미지를 앞서 만든 tool 창에 띄웁니다.
    cv2.waitKey() # 사용자가 아무 키를 누르기 전 까지 이미지를 띄웁니다. 키를 누르면 다음 이미지를 불러옵니다.

import glob # 파일들을 잡기 위한 패키지
import cv2 # 이미지 파일을 처리하기 위한 패키지
import numpy as np

BLOCK = 80 # 한 칸의 크기 입니다. 가로 세로 80픽셀을 의미합니다.
ROW = 9 # 행이 9칸이란 뜻입니다. 영상 높이가 720 픽셀인 경우 이를 블록사이즈(80)으로 나눈 값입니다.
COL = 16 # 열이 16칸이란 뜻입니다. 영상 너비가 1280 픽셀인 경우 이를 블록사이즈(80)으로 나눈 값입니다.

grid = np.zeros([ROW, COL], dtype=np.int) # 9 x 16 2차원 배열을 만들어서 값을 0으로 채웁니다.

img_files = glob.glob('./data/*.jpg') # data폴더 안에 있는 모든 jpg 파일들 목록을 가져옵니다.

cv2.namedWindow('tool') # tool이란 윈도우(창)을 하나 띄웁니다.

for img_file in img_files: # 모든 jpg 파일들 목록에서 jpg 파일명을 하나하나씩 가지고 옵니다.
    img = cv2.imread(img_file) # 가지고 온 jpg 파일명으로부터 이미지를 불러옵니다.

    for r in range(ROW): # 행을 하나씩 선택합니다.
        for c in range(COL): # 열을 하나씩 선택합니다. 
            cv2.rectangle(img, (c*BLOCK, r*BLOCK), ((c+1)*BLOCK, (r+1)*BLOCK), (255, 255, 255), cv2.FILLED * grid[r][c]) # 네모를 그립니다.

    cv2.imshow('tool', img) # 불러온 이미지를 앞서 만든 tool 창에 띄웁니다.
    cv2.waitKey() # 사용자가 아무 키를 누르기 전 까지 이미지를 띄웁니다. 키를 누르면 다음 이미지를 불러옵니다.

import glob # 파일들을 잡기 위한 패키지
import cv2 # 이미지 파일을 처리하기 위한 패키지
import numpy as np

BLOCK = 80 # 한 칸의 크기 입니다. 가로 세로 80픽셀을 의미합니다.
ROW = 9 # 행이 9칸이란 뜻입니다. 영상 높이가 720 픽셀인 경우 이를 블록사이즈(80)으로 나눈 값입니다.
COL = 16 # 열이 16칸이란 뜻입니다. 영상 너비가 1280 픽셀인 경우 이를 블록사이즈(80)으로 나눈 값입니다.

grid = np.zeros([ROW, COL], dtype=np.int) # 9 x 16 2차원 배열을 만들어서 값을 0으로 채웁니다.

def click(event, x, y, flags, param): # 사용자가 마우스를 클릭했을 때 이 함수가 호출됩니다.
    if event == cv2.EVENT_LBUTTONDOWN: # 왼쪽 마우스를 클릭했을 때
        grid[y // BLOCK][x // BLOCK] = (grid[y // BLOCK][x // BLOCK] + 1) % 2 # 해당 그리드 칸이 0이라면 1로, 1이라면 0으로 만듭니다.

img_files = glob.glob('./data/*.jpg') # data폴더 안에 있는 모든 jpg 파일들 목록을 가져옵니다.

cv2.namedWindow('tool') # tool이란 윈도우(창)을 하나 띄웁니다.
cv2.setMouseCallback('tool', click) # tool 창에 마우스 클릭했을 때, click 함수를 호출하도록 설정합니다.

for img_file in img_files: # 모든 jpg 파일들 목록에서 jpg 파일명을 하나하나씩 가지고 옵니다.
    img = cv2.imread(img_file) # 가지고 온 jpg 파일명으로부터 이미지를 불러옵니다.
    grid.fill(0) # 이미자가 바뀌면 흰색 칠 했던 정보도 초기화 합니다.

    while True: # 스페이스 키를 누르기 전 까지는 계속 이미지를 표출합니다. 무한 반복해서 그리기 때문에 마우스로 클릭하면 그때 그때 바뀐 정보로 그려집니다.

        visual = img.copy() # 매번 img 이미지을 복제해서 visual 이미지로 복제합니다.

        for r in range(ROW): # 행을 하나씩 선택합니다.
            for c in range(COL): # 열을 하나씩 선택합니다. 
                cv2.rectangle(visual, (c*BLOCK, r*BLOCK), ((c+1)*BLOCK, (r+1)*BLOCK), (255, 255, 255), cv2.FILLED * grid[r][c]) # visual 이미지에 네모를 그립니다.

        cv2.imshow('tool', visual) # 사각형을 그린 이미지를 앞서 만든 tool 창에 띄웁니다.

        if cv2.waitKey(2) == 32:  # 사용자가 스페이스 키를 눌렸을 때 다음 이미지를 불러옵니다. 그렇지 않다면 2밀리초 이후에 다시 이미지를 그립니다. 
            break

import glob # 파일들을 잡기 위한 패키지
import cv2 # 이미지 파일을 처리하기 위한 패키지
import numpy as np

BLOCK = 80 # 한 칸의 크기 입니다. 가로 세로 80픽셀을 의미합니다.
ROW = 9 # 행이 9칸이란 뜻입니다. 영상 높이가 720 픽셀인 경우 이를 블록사이즈(80)으로 나눈 값입니다.
COL = 16 # 열이 16칸이란 뜻입니다. 영상 너비가 1280 픽셀인 경우 이를 블록사이즈(80)으로 나눈 값입니다.

grid = np.zeros([ROW, COL], dtype=np.int) # 9 x 16 2차원 배열을 만들어서 값을 0으로 채웁니다.

def click(event, x, y, flags, param): # 사용자가 마우스를 클릭했을 때 이 함수가 호출됩니다.
    if event == cv2.EVENT_LBUTTONDOWN: # 왼쪽 마우스를 클릭했을 때
        grid[y // BLOCK][x // BLOCK] = (grid[y // BLOCK][x // BLOCK] + 1) % 2 # 해당 그리드 칸이 0이라면 1로, 1이라면 0으로 만듭니다.

img_files = glob.glob('./data/*.jpg') # data폴더 안에 있는 모든 jpg 파일들 목록을 가져옵니다.

cv2.namedWindow('tool') # tool이란 윈도우(창)을 하나 띄웁니다.
cv2.setMouseCallback('tool', click) # tool 창에 마우스 클릭했을 때, click 함수를 호출하도록 설정합니다.

for img_file in img_files: # 모든 jpg 파일들 목록에서 jpg 파일명을 하나하나씩 가지고 옵니다.
    img = cv2.imread(img_file) # 가지고 온 jpg 파일명으로부터 이미지를 불러옵니다.
    grid.fill(0) # 이미자가 바뀌면 흰색 칠 했던 정보도 초기화 합니다.

    while True: # 스페이스 키를 누르기 전 까지는 계속 이미지를 표출합니다. 무한 반복해서 그리기 때문에 마우스로 클릭하면 그때 그때 바뀐 정보로 그려집니다.

        visual = img.copy() # 매번 img 이미지을 복제해서 visual 이미지로 복제합니다.

        for r in range(ROW): # 행을 하나씩 선택합니다.
            for c in range(COL): # 열을 하나씩 선택합니다. 
                cv2.rectangle(visual, (c*BLOCK, r*BLOCK), ((c+1)*BLOCK, (r+1)*BLOCK), (255, 255, 255), cv2.FILLED * grid[r][c]) # visual 이미지에 네모를 그립니다.

        cv2.imshow('tool', visual) # 사각형을 그린 이미지를 앞서 만든 tool 창에 띄웁니다.

        if cv2.waitKey(2) == 32:  # 사용자가 스페이스 키를 눌렸을 때 다음 이미지를 불러옵니다. 그렇지 않다면 2밀리초 이후에 다시 이미지를 그립니다. 
            flat = np.array(grid).flatten() # 2차원의 그리드 정보를 1차원으로 바꿉니다.
            fp_label = open('annotation.txt', 'a') # annotation.txt 파일을 추가쓰기 권한으로 엽니다.
            fp_label.writelines(img_file + ',' + ''.join(map(str, flat))+'\n') # 1차원으로 변환된 그리드 정보를 이쁘게 씁니다.
            fp_label.close() # 파일을 닫습니다.
            break