blob: 106b5c7104567f2ec10960db479fa46d041dab28 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
# 1、导入模块
import os
import paddlehub as hub
import numpy as np
from collections import Counter
import cv2
import matplotlib.pyplot as plt
path = './data/'
files = os.listdir(path)
imgs = []
for i in files:
imgs.append(path + i)
cv2.imshow('', cv2.imread(imgs[0]))
# 2、加载模型
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')
results = humanseg.segmentation(data={'image':imgs}, visualization=True)
unique, counts = np.unique(results[0]['data'], return_counts=True)
print(dict(zip(unique, counts)))
print()
|
