Commit faf2e178 authored by Maduranga Jayakody's avatar Maduranga Jayakody

import model

parent 6a387603
# IT18255720
#import libries
from imageai.Detection import ObjectDetection, VideoObjectDetection
import sys
import os
import cv2
import time
from shapely.geometry import box
THRESHOLD_ITEM = 0.5
THRESHOLD_PERSON = 0.5
print("Sharp Object Detection Begins")
detector = VideoObjectDetection()
detector2 = VideoObjectDetection()
detector.setModelTypeAsYOLOv3()
detector.setModelPath("../model/yolo.h5")
detector.loadModel()
detector2.setModelTypeAsRetinaNet()
detector2.setModelPath("model/resnet50_coco_best_v2.1.0.h5")
detector2.loadModel()
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment