paddle目标检测cpu 和gpu识别内容差别
收藏
paddle目标检测cpu 和gpu识别内容差别太大了
我用gpu训练目标检测,同样的模型,同样的图片,
import paddlex as pdx import os os.environ['CUDA_VISIBLE_DEVICES'] = '1' model = pdx.load_model('yolov3_mobilenetv1/best_model') result = model.predict('colorF2.png') print(result) pdx.det.visualize('colorF2.png', result, save_dir='./', threshold=0.1)
0
收藏
请登录后评论
您好,请问这个检测模型是用什么方式训练出来的呢?PaddleX嘛?