我一定在使用 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)