Homec4science
Diffusion solarPV (master) Browse

solarPV_identification/labelling_tool/move.py
0d94abfd5245master

Last Change
Raw File

move.py
View Options

import cv2
import numpy as np
import sys
import shutil
import os
class Move():
def __init__(self, image_name, output_dir_name):
self.image_name = image_name
self.output_dir_name = output_dir_name
self.image = cv2.imread(self.image_name,-1)
self.copied_image = self.image.copy()
def do_move(self):
cv2.namedWindow("MOVE", cv2.WINDOW_NORMAL) ## Magnifying the window for more precise labelling
cv2.resizeWindow("MOVE", 1000, 1000) ## 1250
self.move_im = False
while True:
cv2.imshow("MOVE",self.image)
#cv2.setMouseCallback("CROP",self.crop_for_mouse)
keypress = cv2.waitKey(1)
if keypress == ord('y'):
#move to the next file
self.move_im = True
break
if keypress == ord('n'):
self.move_im = False
break
if self.move_im is True:
image_path = self.image_name.split(os.path.sep)
image_dest_path = self.output_dir_name + '/PV/' + image_path[-1]
print("Copying original image ", self.image_name, " to ", image_dest_path)
shutil.copy(self.image_name, image_dest_path)
label_path = os.sep.join(image_path[:-1]) + '/labels/'\
+ image_path[-1][:-4] + '_label' + \
image_path[-1][-4:]
label_dest_path = self.output_dir_name + '/PV/labels/'\
+ image_path[-1][:-4] + '_label' + \
image_path[-1][-4:]
print("Copying labelled image ", label_path, " to ", label_dest_path)
shutil.copy(label_path, label_dest_path)
else:
image_path = self.image_name.split(os.path.sep)
image_dest_path = self.output_dir_name + '/almostPV/' + image_path[-1]
print("Copying original image ", self.image_name, " to ", image_dest_path)
shutil.copy(self.image_name, image_dest_path)
label_path = os.sep.join(image_path[:-1]) + '/labels/'\
+ image_path[-1][:-4] + '_label' + \
image_path[-1][-4:]
label_dest_path = self.output_dir_name + '/almostPV/labels/'\
+ image_path[-1][:-4] + '_label' + \
image_path[-1][-4:]
print("Copying labelled image ", label_path, " to ", label_dest_path)
shutil.copy(label_path, label_dest_path)
c4science · Help