1.问题:
运行train.py时显示通道数错误问题RuntimeError: Given groups=1, weight of size 64 4 7 7, expected input[1, 5, 206, 206] to have 4 channels, but got 512 channels instead。
解决方法:
weights的default中不设置参数,cfg的default需要设置参数。
修改之前的代码:
修改之后的代码:
问题及解决办法:
运行train.py时显示通道数错误问题RuntimeError: Given groups=1, weight of size 64 4 7 7, expected input[1, 5, 206, 206] to have 4 channels, but got 512 channels instead。
weights的default中不设置参数,cfg的default需要设置参数。
修改之前的代码:
修改之后的代码:
问题及解决办法: