小编mjw*_*ock的帖子

分割任务中的掩码值为 0 或 255。我该如何解决?

我一定在使用 fast-ai 库时遇到了严重问题,因为我似乎是唯一遇到此问题的人。每次我尝试学习率查找器或训练网络时,它都会给我一个错误。我花了一周时间才产生这个特定的错误信息,这让我检查了掩码值。事实证明,背景像素为 0,前景像素为 255。这是一个问题,因为我只有两个类。如何在我的 Databunch 对象中将 255 个值更改为 1?有没有办法将每个掩码值除以 255 或者我是否需要事先以某种方式进行?我有点迷失在这个过程中。

这是我收到的错误消息:

LR Finder is complete, type {learner_name}.recorder.plot() to see the graph.

---------------------------------------------------------------------------
Run Code Online (Sandbox Code Playgroud)
LR Finder is complete, type {learner_name}.recorder.plot() to see the graph.

---------------------------------------------------------------------------
Run Code Online (Sandbox Code Playgroud)

这也是我设置数据的方式:

    RuntimeError                              Traceback (most recent call last)

    <ipython-input-20-c7a9c29f9dd1> in <module>()
    ----> 1 learn.lr_find()
          2 learn.recorder.plot()

    8 frames

    /usr/local/lib/python3.6/dist-packages/fastai/train.py in lr_find(learn, start_lr, end_lr, num_it, stop_div, wd)
         39     cb = LRFinder(learn, start_lr, end_lr, num_it, stop_div)
         40     epochs = int(np.ceil(num_it/len(learn.data.train_dl)))
    ---> 41     learn.fit(epochs, start_lr, callbacks=[cb], wd=wd)
         42 
         43 …
Run Code Online (Sandbox Code Playgroud)

mask image-segmentation pytorch fast-ai

6
推荐指数
1
解决办法
1034
查看次数

标签 统计

fast-ai ×1

image-segmentation ×1

mask ×1

pytorch ×1