如何自动生成图片用于测试 pytorch（No image? No need image）

Posted 2021-02-15 ziytong

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了如何自动生成图片用于测试 pytorch（No image? No need image）相关的知识，希望对你有一定的参考价值。

if __name__ == ‘__main__‘:
    module = CAM_Module()
    in_data = torch.randint(0, 255, (2, 3, 7, 7), dtype=torch.float32)

以上是关于如何自动生成图片用于测试 pytorch（No image? No need image）的主要内容，如果未能解决你的问题，请参考以下文章