PyTorch上でiouをimportする方法が分かりません

Question

前提

ここに質問の内容を詳しく書いてください。
（例）
pytorch_lightningを用いて画像分類の学習をしています。
評価のためにiouを用いるためにtorchmetricsからiouをimportしようとしても以下のようなエラーメッセージが出てしまいます。
初歩的な質問で申し訳ございませんが、宜しくお願い致します。

実現したいこと

pytorch_lightningを用いてiouによる評価を行いたいです。

発生している問題・エラーメッセージ

ImportError  Traceback (most recent call last)
<ipython-input-34-87122a19004f> in <module>()
      1 get_ipython().system('pip install -q torchmetrics')
----> 2 from torchmetrics.functional import accuracy, iou

ImportError: cannot import name 'iou' from 'torchmetrics.functional' (/usr/local/lib/python3.7/dist-packages/torchmetrics/functional/__init__.py)

---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------

該当のソースコード

!pip install -q torchmetrics
from torchmetrics.functional import accuracy, iou

試したこと

以前はimportできていたと思うので、versionの問題もあるかと考えて古いversionにするなど試みましたが上手くいきません。

補足情報（FW/ツールのバージョンなど）

以前のバージョンがいくつだったのかは申し訳ございませんが正確には分かりません。
今年5月ごろに別の問題を学習している際にはfrom torchmetrics.functional import iouでimportできていた記録がありますので、その時にはversionの指定は特に行なっていませんでした。
Pytorch_lightningの現在のversionは1.7.1になります。
宜しくお願い致します。

Comments

[meg_]

> 以前はimportできていたと思うので、versionの問題もあるかと考えて古いversionにするなど試みましたが上手くいきません。
以前のバージョンと現在のバージョンは何でしょうか？質問に追記してください。

[programbegin]

ご連絡ありがとうございます。
以前のバージョンがいくつだったのかは申し訳ございませんが正確には分かりません。
今年5月ごろに別の問題を学習している際にはfrom torchmetrics.functional import iouでimportできていた記録がありますので、その時にはversionの指定は特に行なっていませんでした。
Pytorch_lightningの現在のversionは1.7.1になります。

[meg_]

> 質問に追記してください。
質問は編集可能ですので質問に追記しましょう。（他の回答者に見てもらえやすくするためです）

Answer 1

cannot import name 'iou' from 'torchmetrics.functional'

torchmetrics.functional.iouは、torchmetricsの
0.7.3
にはありますが、
0.8.0
にはありません

　
【追記】
torchmetrics.functional.iou
から
torchmetrics.functional.jaccard_index
に、名前が変わったようです
rename IoU -> Jaccard Index #662

Comments

[programbegin]

早々にご回答いただきまして誠にありがとうございます。
!pip install -q torchmetrics==0.7.3
import torchmetrics
from torchmetrics.functional import accuracy, iou
上記のように記述してimport試みましたが、以下のような同じエラーメッセージが表示されてしまいます。他のversionの問題なのでしょうか？
ImportError: cannot import name 'iou' from 'torchmetrics.functional' (/usr/local/lib/python3.7/dist-packages/torchmetrics/functional/__init__.py)

[jbpb0]

torchmetrics.functional.iou
から
torchmetrics.functional.jaccard_index
に、名前が変わったようです
https://github.com/Lightning-AI/metrics/pull/662/commits/2e96b03de0e0b3a8ddb5010df1df9f5e3197238b

[jbpb0]

> cannot import name 'iou' from 'torchmetrics.functional' (/usr/local/lib/python3.7/dist-packages/torchmetrics/functional/__init__.py)

!pip install -q torchmetrics==0.7.3

を実行した後に、

/usr/local/lib/python3.7/dist-packages/torchmetrics/functional/__init__.py
の内容が、
https://github.com/Lightning-AI/metrics/blob/v0.7.3/torchmetrics/functional/__init__.py
と全く同じか、確認してください

全く同じなら、下記の行が30行目にあります

from torchmetrics.functional.classification.iou import iou # noqa: F401

[jbpb0]

!pip show torchmetrics

を実行してください
いろいろ表示される中に「Location:」で始まる行がありますが、そこに書かれてるパスは下記でしょうか？

/usr/local/lib/python3.7/dist-packages

もし違うなら、pipで入れたものと、pythonが使ってるものが、別のものです

[programbegin]

お忙しい中、詳しいご回答ありがとうございます。
仰るようにfrom torchmetrics.functional import jaccard_indexでエラーになりませんでした！
また!pip show torchmetricsでは「/usr/local/lib/python3.7/dist-packages」と表示されました。
jaccard_indexでこれからIoUの計算を行ってみたいと思います。
とても勉強になりました。誠にありがとうございます。

[jbpb0]

> !pip show torchmetricsでは「/usr/local/lib/python3.7/dist-packages」と表示されました。

ならば、

!pip install -q torchmetrics==0.7.3

を実行してインストールが成功したら、
/usr/local/lib/python3.7/dist-packages/torchmetrics/functional/__init__.py
の内容は
https://github.com/Lightning-AI/metrics/blob/v0.7.3/torchmetrics/functional/__init__.py
と同じになり、下記のエラーにはならないはずなのですが、エラーになったのが不思議です

> ImportError: cannot import name 'iou' from 'torchmetrics.functional' (/usr/local/lib/python3.7/dist-packages/torchmetrics/functional/__init__.py)

Answer 2

torchmetrics 0.6.0ではfrom torchmetrics.functional import accuracy, iouを実行することができました。

Comments

[programbegin]

早々にご回答いただきまして誠にありがとうございます。
!pip install -q torchmetrics==0.6.0
import torchmetrics
from torchmetrics.functional import accuracy, iou
上記のように記述してimport試みましたが、以下のような同じエラーメッセージが表示されてしまいます。他のversionの問題なのでしょうか？
ImportError: cannot import name 'iou' from 'torchmetrics.functional' (/usr/local/lib/python3.7/dist-packages/torchmetrics/functional/__init__.py)

[meg_]

torchmetrics.__version__ で 0.6.0 と表示されますか？