使用opencv讀圖片
格式【h*w*c】
轉爲【c*h*w】
img = cv2.imread('../imgs/86.jpg',)
img=numpy.transpose(img,(2,0,1))
input=torch.from_numpy(img)
input=torch.tensor(input,dtype=torch.float)
使用opencv讀圖片
格式【h*w*c】
轉爲【c*h*w】
img = cv2.imread('../imgs/86.jpg',)
img=numpy.transpose(img,(2,0,1))
input=torch.from_numpy(img)
input=torch.tensor(input,dtype=torch.float)
FPN class FPN(nn.Module): def __init__(self, block, layers): super(FPN, self).__init__() self.