高分辨率图像的目标检测推理在CPU上花费大量时间

时间:2019-06-09 10:33:04

标签: pytorch fast-ai

我已经在图像大小为224的pascal数据集上训练了ML模型,但是在推断新图像(某些分辨率更高,而某些分辨率比pascal图像更高)时,我得到了{ 1}}

pil2tensor()

对于小尺寸图像(例如Google的一些低分辨率图像),ML模型可以在几秒钟内正确推断,但对于稍高分辨率的图像,则需要20-40分钟!!

1 个答案:

答案 0 :(得分:0)

解决了该问题,这是正确的代码:

class AbstractC(ABC):

   def a(self):
       raise NotImplementedError('Implement _a_ method')

class ConcreteC(AbstractC):
   def __init__(self, name, *args, **kwargs):
       super().__init__(*args, **kwargs)
       self.a = name