実現させたい事
下2つのサイトを参照し画像処理のコードを実行してみたが、モジュールの部分でエラーになってしまうので、そのエラーを解消させたいと考えております。
https://tech.fusic.co.jp/posts/2020-01-20-remove-background/
https://www.hamlet-engineer.com/posts/remove_bg02.html
環境
windows10
python 3.7.9
やったこと
python
1import numpy as np 2import cv2 3import matplotlib.pyplot as plt 4 5import torch 6import torchvision 7from torchvision import transforms 8 9# 画像の読み込み 10image_path = r"D:\test\test.png" 11img = cv2.imread(image_path) 12img = img[...,::-1] #BGR->RGB 13h,w,_ = img.shape 14img = cv2.resize(img,(320,320)) 15 16device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 17 18model = torchvision.models.segmentation.deeplabv3_resnet101(pretrained=True) 19model = model.to(device) 20model.eval(); 21 22preprocess = transforms.Compose([ 23 transforms.ToTensor(), 24 transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), 25]) 26 27input_tensor = preprocess(img) 28input_batch = input_tensor.unsqueeze(0).to(device) 29 30with torch.no_grad(): 31 output = model(input_batch)['out'][0] 32output = output.argmax(0) 33mask = output.byte().cpu().numpy() 34mask = cv2.resize(mask,(w,h)) 35img = cv2.resize(img,(w,h)) 36plt.gray() 37plt.figure(figsize=(20,20)) 38plt.subplot(1,2,1) 39plt.imshow(img) 40plt.subplot(1,2,2) 41plt.imshow(mask);
を実行すると
File "d:/test_test/test.py", line 6, in <module> import torchvision File "C:\Users\81809\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\__init__.py", line 7, in <module> from torchvision import models File "C:\Users\81809\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\models\__init__.py", line 19, in <module> from . import segmentation File "C:\Users\81809\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\models\segmentation\__init__.py", line 2, in <module> from .deeplabv3 import * File "C:\Users\81809\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\models\segmentation\deeplabv3.py", line 185 pretrained: bool = False, ^ SyntaxError: invalid syntax
となってしまいます。
調べた事
torchvision pretrained: bool = False, SyntaxError: invalid syntax
python モジュール エラー
python モジュール 中身 上書き
などでググってみましたが解決できませんでしたので、こちらで質問させていただきました。
質問
モジュールの中身のコードを書き換えればいいのでしょうか??
どうしたらこのエラーを解消できますでしょうか??どなたか有識者様のお力をお借りできればと思います。宜しくお願い致します。
2022/5/29/21:28 追記
コードとエラー内容を追記しました。

回答1件
あなたの回答
tips
プレビュー