​ I am seeing an error, when trying to access camera with pyroch in google co lab?

[tedzz12]

Hai i am trying to detect objects through laptop camera using PyTorch in google co lab, when i run cv2 code, i am seeing an error, can anyone help me to solve the problem, here is the code:-

import cv2
import torch
import torchvision.transforms as transforms
from torchvision.models import resnet50

model = model_x_5
model.eval()
transform = transforms.Compose([
transforms.ToPILImage(),
transforms.Resize(size=(64,64)),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485,0.456,0.406], std=[0.229,0.224,0.225])
])
cap = cv2.VideoCapture(0)

while True:
ret,frame = cap.read()
frame_rgb = cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
input_tensor = transform(frame_rgb)
input_tensor = input_tensor.unsqueeze(0)
with torch.no_grad():
output = model(input_tensor)
predicted_idx = torch.max(output,1)
labels = [“Toothbrush”]
predicted_label = labels[predicted_idx.item()]
cv2.putText(frame,predicted_label,(10,30),cv2.FONT_HERSHEY_SIMPLEX,1,(0,255,0),2)
cv2.imshow(“object detection”,frame)
if cv2.waitKey(1) & 0xFF == ord(‘q’):
break
cap.release()
cv2.destroyAllWindows()

error:----------------------------------------------------------------------------
error Traceback (most recent call last)
in <cell line: 17>()
17 while True:
18 ret,frame = cap.read()
—> 19 frame_rgb = cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
20 input_tensor = transform(frame_rgb)
21 input_tensor = input_tensor.unsqueeze(0)

error: OpenCV(4.7.0) /io/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function ‘cvtColor’

[mrdbourke]

Hi @tedzz12 ,

I don't have much experience with cv2 but what have you tried so far?

Have you seen this thread on Stack Overflow here: https://stackoverflow.com/questions/52676020/imread-returns-none-violating-assertion-src-empty-in-function-cvtcolor-er

I'm not sure how much I can help troubleshoot since this discussion forum is mostly about code related to the course.

[tedzz12]

okay, thank you

[tedzz12]

i tried to detect some basic objects like toothbrush,cellphone to detect through webcam.with CNN