部署pytorch.孔型python脚本

0

的问题

在成功的培训我yolact模型利用一个自定义的数据集,我很高兴与推断结果的输出的通过eval.py 使用这种命令,从蟒蛇的终端:

python eval.py --trained_model=./weights/yolact_plus_resnet50_abrasion_39_10000.pth --config=yolact_resnet_abrasion_config --score_threshold=0.8 --top_k=15 --images=./images:output_images

现在我要运行这一推断,从我自己的蟒蛇脚本,而不是使用蟒蛇的终端。 我想要能够得到边界箱检测上的摄像头框架获得通过,这代码如下。 任何想法?

import cv2

src = cv2.VideoCapture(0)

while True:
    ret, frame = src.read()
    cv2.imshow('frame', frame)
    key = cv2.waitKey(5)
    if key == (27):
        break

的eval.py 代码是在这里,在Yolact库 https://github.com/dbolya/yolact/blob/master/eval.py

computer-vision opencv python pytorch
2021-11-19 17:21:59
1

最好的答案

0

我只是写的代码这里你。

步骤1:尽量载入模型使用的线路从 这里 结束 这里

步骤2:使用这一 功能 进行评估。 而不是cv2.imread,你只需要送你 框架

步骤3:按照这一 功能 得到边界的箱子。 尤其是这个 线的. 只是引用的't'可变的,你会得到你的边界框。

希望这有所帮助。 让我知道如果你需要更多的澄清。

2021-11-20 08:35:20

以下thoose步骤,我可以修改eval.py 并获得评估和边界的箱子。 非常感谢你的回答,非常有帮助:D
Ramiro Sena

其他语言

此页面有其他语言版本

Русский
..................................................................................................................
Italiano
..................................................................................................................
Polski
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
हिन्दी
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
Português
..................................................................................................................
ไทย
..................................................................................................................
Español
..................................................................................................................
Slovenský
..................................................................................................................