So this was a fun project! I’ve created a watermark on a live video.
Demo:
Things Required:
- Jetson Nano
- Keyboard
- Mouse
- HDMI Screen
- Raspberry Pi Camera
- External Power Supply
Reference Material:
- You can check out this video for the project: (LINK)
Code:
Note that code below is in python
View on/Download from Github: (LINK)
import cv2
dispW= 320
dispH=240
flip=2
#load image
cvLogo= cv2.imread('/home/aarushi/Desktop/pyPro/cv.jpg')
cvLogo= cv2.resize(cvLogo, (320,240))
cv2.imshow('Original', cvLogo)
cv2.moveWindow('Original', 0, 300)
#convert cvLogo to gray for thresholding
cvLogoGray= cv2.cvtColor(cvLogo, cv2.COLOR_BGR2GRAY)
#Threshold
_,BGMask= cv2.threshold(cvLogoGray, 180, 255, cv2.THRESH_BINARY)
FGMask= cv2.bitwise_not(BGMask)
FG= cv2.bitwise_and(cvLogo, cvLogo, mask= FGMask)
camSet= 'nvarguscamerasrc ! video/x-raw(memory:NVMM), width=3264, height=2464, format=NV12, framerate=21/1 ! nvvidconv flip-method='+str(flip)+' ! video/x-raw, width='+str(dispW)+', height='+str(dispH)+', format=BGRx ! videoconvert ! video/x-raw, format=BGR ! appsink'
cam= cv2.VideoCapture(camSet)
while True:
ret, frame= cam.read();
BG= cv2.bitwise_and(frame, frame, mask= BGMask)
Blended= cv2.addWeighted(frame, .8, cvLogo, .2, 0)
FG2= cv2.bitwise_and(Blended, Blended, mask= FGMask)
watermark= cv2.add(BG, FG2)
cv2.imshow('WaterMark', watermark)
cv2.moveWindow('WaterMark', 400, 0)
cv2.imshow('nanoCam', frame)
cv2.moveWindow('nanoCam', 0,0)
if cv2.waitKey(1) == ord('q'):
break
cam.release()
cv2.destroyAllWindows()
Stay Tuned for upcoming posts on Jetson Nano and AI Projects! Want to be the first to know when a new and amazing post comes up?? Then feel free to subscribe!
Happy Learning!!
