运行一个使用pytorch框架的简单分类程序。
运行train.py文件生成.ckpt模型。
运行inference.py文件来测试模型,模型的路径和待测试的图片的目录由命令行参数给出。
我用cmd运行的,截图如下:
自己写的命令行参数的话argv是从下标1开始的。
但是如果我在settings.json里这样写:
不管是在VS Code里运行还是在cmd运行都会显示数组越界:IndexError: list index out of range.
不晓得是为什么。
VS Code命令行参数
最新推荐文章于 2024-04-17 21:31:09 发布