Question

我目前正在研究一个预先培训的移动网络，我想用包含png图像的数据集重新训练。

我将再培训脚本称为：

super()

看起来图像需要是jpg，是否可以使用png图像？

Answer 1

可以确认它不适用于png文件。但是，我编写了一个bash脚本，将其放置在与数据集的子类相同的目录中时，可以将图像转换为jpg。

首先，您需要通过以下方式安装imagemagick软件包：

sudo apt-get install imagemagick

然后您可以运行以下脚本：

#!/bin/bash

for d in */ ; do
    cd "$d"
    for p in * ; do
        IFS='.' read -r -a array <<< "$p"
        convert "$p" "${array[0]}".jpg
    done
    cd ..
done

编辑：

retrain.py确实有一个带有有效扩展名的列表（第151行）：

extensions = ['jpg', 'jpeg', 'JPG', 'JPEG']

虽然我没有尝试将“ png”添加到列表中

是否有可能用png图像重新训练一个mobilenet神经网络？

1 个答案: