游戏的对象检测

时间:2018-12-09 14:02:06

标签: python object-detection

我想为皮瓣鸟创建一个AI,并希望使用对象检测和神经网络来实现它(我知道这不是最有效的方法,但这是我需要的方法)。 我使用了ImageAI python模块,但实际上并没有完成我想做的事情。 我的代码是:

from imageai.Detection import ObjectDetection
import os

execution_path = os.getcwd()

detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))
detector.loadModel()
detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg"))

for eachObject in detections:
    print(eachObject["name"] , " : " , eachObject["percentage_probability"] )

我给了这张图片,看它是否知道这是一个对象:

pipe image

但是它什么也没做。

所以我的问题是:有什么可以在没有数据库的情况下检测对象的东西吗?

0 个答案:

没有答案