import?os
import?argparse
import?torch

from?torchvision?import?transforms
from?models.fast_scnn?import?get_fast_scnn
from?PIL?import?Image
from?utils.visualize?import?get_color_pallete

parser?=?argparse.ArgumentParser（
????description=‘Predict?segmentation?result?from?a?given?image‘）
parser.add_argument（‘--model‘?type=str?default=‘fast_scnn‘
????????????????????help=‘model?name?（default:?fast_scnn）‘）
parser.add_argument（‘--dataset‘?type=str?default=‘citys‘
????????????????????help=‘dataset?name?（default:?citys）‘）
parser.add_argument（‘--weights-folder‘?default=‘./weights‘
????????????????????help=‘Directory?for?saving?checkpoint?models‘）
parser.add_argument（‘--input-pic‘?type=str
????????????????????default=‘./datasets/citys/leftImg8bit/test/berlin/berlin_000000_000019_leftImg8bit.png‘
????????????????????help=‘path?to?the?input?picture‘）
parser.add_argument（‘--outdir‘?default=‘./test_result‘?type=str
????????????????????help=‘path?to?save?the?predict?result‘）

parser.add_argument（‘--cpu‘?dest=‘cpu‘?action=‘store_true‘）
parser.set_defaults（cpu=False）

args?=?parser.parse_args（）


def?demo（）:
????device?=?torch.device（“cuda“?if?torch.cuda.is_available（）?else?“cpu“）
????#?output?folder
????if?not?os.path.exists（args.outdir）:
????????os.makedirs（args.outdir）

????#?image?transform
????transform?=?transforms.Compose（[
????????transforms.ToTensor（）
????????transforms.Normalize（[0.485?0.456?0.406]?[0.229?0.224?0.225]）
????]）
????image?=?Image.open（args.input_pic）.convert（‘RGB‘）
????image?=?transform（image）.unsqueeze（0）.to（device）
????model?=?get_fast_scnn（args.dataset?pretrained=True?root=args.weights_folder?map_cpu=args.cpu）.to（device）
????print（‘Finished?loading?model!‘）
????model.eval（）
????with?torch.no_grad（）:
????????outputs?=?model（image）
????pred?=?torch.argmax（outputs[0]?1）.squeeze（0）.cpu（）.data.numpy（）
????mask?=?get_color_pallete（pred?args.dataset）
????outname?=?os.path.splitext（os.path.split（args.input_pic）[-1]）[0]?+?‘.png‘
????mask.save（os.path.join（args.outdir?outname））


if?__name__?==?‘__main__‘:
????demo（）

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2019-05-13?13:40??Fast-SCNN-pytorch-master\
?????文件????????1358??2019-05-13?13:40??Fast-SCNN-pytorch-master\.gitignore
?????文件???????11357??2019-05-13?13:40??Fast-SCNN-pytorch-master\LICENSE
?????文件????????3058??2019-05-13?13:40??Fast-SCNN-pytorch-master\README.md
?????目錄???????????0??2019-05-13?13:40??Fast-SCNN-pytorch-master\data_loader\
?????文件?????????212??2019-05-13?13:40??Fast-SCNN-pytorch-master\data_loader\__init__.py
?????文件????????8220??2019-05-13?13:40??Fast-SCNN-pytorch-master\data_loader\cityscapes.py
?????文件????????2083??2019-05-13?13:40??Fast-SCNN-pytorch-master\demo.py
?????文件????????2185??2019-05-13?13:40??Fast-SCNN-pytorch-master\eval.py
?????目錄???????????0??2019-05-13?13:40??Fast-SCNN-pytorch-master\models\
?????文件???????????0??2019-05-13?13:40??Fast-SCNN-pytorch-master\models\__init__.py
?????文件????????9263??2019-05-13?13:40??Fast-SCNN-pytorch-master\models\fast_scnn.py
?????目錄???????????0??2019-05-13?13:40??Fast-SCNN-pytorch-master\png\
?????文件??????116083??2019-05-13?13:40??Fast-SCNN-pytorch-master\png\Fast-SCNN.png
?????文件???????31427??2019-05-13?13:40??Fast-SCNN-pytorch-master\png\frankfurt_000001_058914_gtFine_color.png
?????文件?????2230101??2019-05-13?13:40??Fast-SCNN-pytorch-master\png\frankfurt_000001_058914_leftImg8bit.png
?????文件???????22325??2019-05-13?13:40??Fast-SCNN-pytorch-master\png\frankfurt_000001_058914_seg.png
?????文件????????8669??2019-05-13?13:40??Fast-SCNN-pytorch-master\train.py
?????目錄???????????0??2019-05-13?13:40??Fast-SCNN-pytorch-master\utils\
?????文件???????????0??2019-05-13?13:40??Fast-SCNN-pytorch-master\utils\__init__.py
?????文件????????4908??2019-05-13?13:40??Fast-SCNN-pytorch-master\utils\loss.py
?????文件????????3270??2019-05-13?13:40??Fast-SCNN-pytorch-master\utils\lr_scheduler.py
?????文件????????6385??2019-05-13?13:40??Fast-SCNN-pytorch-master\utils\metric.py
?????文件????????3919??2019-05-13?13:40??Fast-SCNN-pytorch-master\utils\visualize.py
?????目錄???????????0??2019-05-13?13:40??Fast-SCNN-pytorch-master\weights\
?????文件?????4710241??2019-05-13?13:40??Fast-SCNN-pytorch-master\weights\fast_scnn_citys.pth