-
在models目录下创建一个新的.py文件,比如说my_module.py。
-
在my_module.py中定义你的模块。可以参考原有的模块(如yolo.py、anchor.py等)的写法进行代码编写。这样,你就可以使用'--model my_module'来训练你定义的模型了。
-
在train.py中添加相应的命令行解析语句,以支持用户在训练时选择使用你的自定义模块。
-
在train.py中添加相应的命令行解析语句,以支持用户在训练时选择使用你的自定义模块
-
最后,在训练时只需要使用你的模型名称就可以启用你的自定义模块