PytorchによるUNetが実行できない

Question

python train.py
とコマンドを入れても、エラー文が出てきます。
どなたか教えてください。

ーエラー文ー
Traceback (most recent call last):
File "train.py", line 4, in <module>
from UNet import model
ModuleNotFoundError: No module named 'UNet

ファイルの場所がおかしいのかと思って確認しても、特に間違っていない。

--フォルダ--
UNet
-dataset
-test
-config.yaml
-model.py
-prediction.py
-prepare_data.py
-train.py

train.py

import tensorflow as tf
import tensorflow.keras.backend as K
import yaml
from UNet import model
from UNet.prepare_data import data_gen


def dice_coef(y_true, y_pred):
    y_true = K.flatten(y_true)
    y_pred = K.flatten(y_pred)
    intersection = K.sum(y_true * y_pred)
    return 2.0 * intersection / (K.sum(y_true) + K.sum(y_pred) + 1)


def dice_coef_loss(y_true, y_pred):
    return 1.0 - dice_coef(y_true, y_pred)


def train(args: dict):
    lr: float = args["learning_rate"]
    n_classes: int = args["n_classes"]

    unet = model.UNet(args)
    unet.compile(
        optimizer=tf.keras.optimizers.Adam(lr),
        loss=dice_coef_loss,
        metrics=["accuracy"],
    )
    unet.summary()

    ckpt = tf.keras.callbacks.ModelCheckpoint(
        filepath="./UNet/params/model.h5",
        monitor="loss",
        save_best_only=True,
        save_weights_only=True,
        verbose=1,
    )

    segmented_data = args[
        "segmented_data"
    ]  # os.path.join(args.train_data, "../segmented_images")
    generator = data_gen(
        args["train_data"], segmented_data, args["batch_size"], n_classes
    )
    unet.fit_generator(generator, steps_per_epoch=30, epochs=100, callbacks=[ckpt])


if __name__ == "__main__":
    if tf.__version__ >= "2.0.0":
        device = tf.config.experimental.list_physical_devices("GPU")
        if len(device) > 0:
            for dev in device:
                tf.config.experimental.set_memory_growth(dev, True)
    else:
        config = tf.ConfigProto()
        config.gpu_options.allow_growth = True
        tf.keras.backend.set_session(tf.Session(config=config))

    with open("./UNet/config.yaml") as f:
        args = yaml.safe_load(f)

    train(args)

Comments

[jbpb0]

> No module named 'UNet

Pythonで

import os
print(os.listdir("./"))

を実行した時の結果表示の中に、「UNet」が含まれていますでしょうか？

[jbpb0]

・上記の結果表示の中に、「UNet」が含まれている
・「UNet」ディレクトリの中に「model.py」がある
という状態になれば、

> from UNet import model

でエラー出なくなると思います

> --フォルダ--
UNet

以降に書かれてるのを見ると、「model.py」があるのは、「UNet」ディレクトリの中ではないように見えます
・「UNet」ディレクトリの中に、「dataset」ディレクトリと「test」ディレクトリがある
・その「test」ディレクトリの中に、「model.py」とかがある
のですか？

[tiroha]

回答ありがとうございます。
書き方が悪かったです。
・UNetフォルダにフォルダdataset,testがある。
・--となっている部分はフォルダUNetに入っている(testには入っていない)
質問文を修正します。

Answer 1

from UNet import model
from UNet.prepare_data import data_gen

を、

import model
from prepare_data import data_gen

に変更してみると動くかもしれません。

Comments

[tiroha]

変更したところ、その部分はうまくいきました。
他の所でエラー文が出ました。
train(args)のargsを変更ですかね？
/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/training.py:1940: UserWarning: `Model.fit_generator` is deprecated and will be removed in a future version. Please use `Model.fit`, which supports generators.
warnings.warn('`Model.fit_generator` is deprecated and '
Traceback (most recent call last):
File "train.py", line 62, in <module>
train(args)
File "train.py", line 45, in train
unet.fit_generator(generator, steps_per_epoch=30, epochs=100, callbacks=[ckpt])
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/training.py", line 1957, in fit_generator
initial_epoch=initial_epoch)
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/training.py", line 1147, in fit
steps_per_execution=self._steps_per_execution)
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/data_adapter.py", line 1364, in get_data_handler
return DataHandler(*args, **kwargs)
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/data_adapter.py", line 1166, in __init__
model=model)
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/data_adapter.py", line 809, in __init__
peek, x = self._peek_and_restore(x)
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/data_adapter.py", line 866, in _peek_and_restore
peek = next(x)
File "/content/drive/My Drive/UNet-Keras/ML_models-master/UNet/prepare_data.py", line 10, in data_gen
inputs = np.array(os.listdir(train_data))
FileNotFoundError: [Errno 2] No such file or directory: './UNet/dataset/raw_images'

[jbpb0]

> No such file or directory: './UNet/dataset/raw_images'

エラーの原因は、上記ディレクトリがないこと、のようです

[tiroha]

./dataset/raw_imagesにしたところ、エラー文は無くなりました。