运行SSD代码的主函数报错

在调试预测后处理过程中遇到问题，我需要读取预测结果，对预测结果进行疲劳检测的后续判断，于是准备运行网络读取预测标签，出现报错。

######代码
 if __name__ == '__main__':
    ssd1 = SSD()
    x = torch.rand([1,3,300,300])
    y = ssd1(x)
    print(y[0])


###### 报错
Traceback (most recent call last):
  File "D:/疲劳驾驶检测/ssd-pytorch-master/ssd-pytorch-master/SSD.py", line 394, in <module>
    y = ssd1(x)
TypeError: 'SSD' object is not callable

###### 但是我在开头已经定义了网络模型，运行还是会报错

###### 我想要达到的结果