双线性神经网络 林崇宇,Aruni RoyChowdhury,Subhransu Maji提出的的双线性CNN的pytorch实现。 整个模型的框架如下图所示: 在此实现中,基础网络使用resnet34结构 依存关系 python> = 3.5 火炬> = 0.4 另外,请将项目文件夹添加到PYTHONPATH并pip install以下软件包: tqdm 数据 下载图片和注释。 将它们提取到data/raw/ 训练 根据本文的要求,该模型的训练分为以下两个步骤: 预训练模型的参数是固定的,只训练最后一个完整的连接层。 运行以下代码。 python bilinear_ResNet_linear_layer.py 在此步骤中,需要计算图像数据的平均值和方差,以进行图像预处理 模型中的所有参数都经过训练。 运行以下代码。 python bilinear_ResNet_fine_tunin
2021-08-16 10:58:19 117KB pytorch bilinear-cnn Python
1