Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,244 Bytes
59b2a81 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
'''
This repo is provided to change the destination area.
'''
import os, cv2
def draw_dot(ref_img, new_h, new_w):
# Draw the dot
dot_range = 3
for i in range(-1*dot_range, dot_range+1):
for j in range(-1*dot_range, dot_range+1):
dil_vertical, dil_horizontal = new_h + i, new_w + j
if (0 <= dil_vertical and dil_vertical < ref_img.shape[0]) and (0 <= dil_horizontal and dil_horizontal < ref_img.shape[1]):
ref_img[dil_vertical, dil_horizontal, :] = [0, 128, 0]
return ref_img
if __name__ == "__main__":
instance_path = "datasets/validation_thisthat14/000049/"
new_w, new_h = 385, 310
# 256.1850280761719 241.71287155151367
# Read the items
data_path = os.path.join(instance_path, "data.txt")
ref_img_path = os.path.join(instance_path, "im_0.jpg")
ref_img = cv2.imread(ref_img_path)
# Read the first point
file1 = open(data_path, 'r')
Lines = file1.readlines()
frame_idx, horizontal, vertical = Lines[0].split(' ')
ref_img = draw_dot(ref_img, int(float(vertical)), int(float(horizontal)))
# Second dot
ref_img = draw_dot(ref_img, new_h, new_w)
# Store the image
cv2.imwrite("visual.png", ref_img) |