2020-11-23

使用python 对图像中的目标打标签

import cv2
import os
from PIL import Image
from pylab import *
import pandas as pd
import tqdm
import msvcrt

coordinate=[]
for frame in tqdm.tqdm(range(76,1076)):
    frame_str='E:/kongtianbeigbm/PLANEUAV/'+str(frame)+'.png'
    im = array(Image.open(frame_str))
    imshow(im)
    #print('Please click 1 points')
    x =ginput(1)

    coordinate.append([x[0][0],x[0][1]])
    #print('you clicked:',x)
    plt.ioff()  # 显示完后一定要配合使用plt.ioff()关闭交互模式，否则可能出奇怪的问题

    plt.clf()  # 清空图片
    plt.close()  # 清空窗口
    #show()
print(coordinate)
name=['x','y']
target_coordinate=pd.DataFrame(columns=name,data=coordinate)
target_coordinate.to_csv('target_coordinate.csv')