训练没有归一化,预测却有归一化,是不是有问题?
seven-linglx opened this issue · 2 comments
seven-linglx commented
训练时候加载数据是dataloader.py 222行,是没有对图片做mean和std归一化的,但预测时predict.py -> centernet.py -> util/util.py -> preprocess_input里却对图片做了mean、std归一化,这应该有问题吧?
bubbliiiing commented
120行
image = np.transpose(preprocess_input(image), (2, 0, 1))
seven-linglx commented
120行 image = np.transpose(preprocess_input(image), (2, 0, 1))
确实,是我没看清。
对图片的处理统一放在get_random_data函数里是不是会更好些。