Skip to content

Commit

Permalink
Update main.py
Browse files Browse the repository at this point in the history
  • Loading branch information
SoKul-13 committed Mar 2, 2023
1 parent 0259d22 commit a68a9f5
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions v2/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,13 @@
cap.set(4,480)

classNames = []
classFile = r'C:\Users\Soham Kulkarni\OneDrive\Documents\GitHub\vechtor\v2\coco.names.txt'
classFile = r'C:\soham - Personal\Documents\GitHub\vechtor\v2\coco.names.txt'
with open(classFile,'rt') as f:
classNames = f.read().rstrip('n').split('n')

#print(classNames)
configPath = r'C:\Users\Soham Kulkarni\OneDrive\Documents\GitHub\vechtor\v2\ssd_mobilenet_v3_large_coco_2020_01_14.pbtxt'
weightsPath = r'C:\Users\Soham Kulkarni\OneDrive\Documents\GitHub\vechtor\v2\frozen_inference_graph.pb'
configPath = r'C:\soham - Personal\Documents\GitHub\vechtor\v2\ssd_mobilenet_v3_large_coco_2020_01_14.pbtxt'
weightsPath = r'C:\soham - Personal\Documents\GitHub\vechtor\v2\frozen_inference_graph.pb'

net = cv2.dnn_DetectionModel(weightsPath,configPath)
net.setInputSize(320,320)
Expand Down

0 comments on commit a68a9f5

Please sign in to comment.