Spaces:
Build error
Build error
白鹭先生
commited on
Commit
•
7e1fccb
1
Parent(s):
48222b7
修改测试图片
Browse files- image.png +0 -0
- img/01203125-90_266-254&432_452&493-450&492_257&493_254&432_452&433-0_0_3_24_30_31_29_29-158-57.jpg +0 -0
- img/0125-89_86-273&524_472&588-466&592_269&597_270&523_467&518-0_0_18_29_27_30_32-112-29.jpg +0 -0
- img/0125-90_90-348&506_528&586-534&576_352&581_351&512_533&507-0_0_6_25_25_24_24-141-25.jpg +0 -0
- img/02-90_86-287&458_543&538-532&542_295&544_294&457_531&455-0_0_30_12_30_31_30-158-55.jpg +0 -0
- img/02-90_88-223&485_480&573-477&573_243&567_246&489_480&495-0_0_10_30_32_32_24-163-61.jpg +0 -0
- img/2022-04-08 134850.png +0 -0
- img/2022-04-08 134916.png +0 -0
- img/2022-04-08 134931.png +0 -0
- img/2022-04-08 134950.png +0 -0
- img/2022-04-08 155620.png +0 -0
- plate.py +47 -0
- utils/__pycache__/utils.cpython-38.pyc +0 -0
image.png
ADDED
img/01203125-90_266-254&432_452&493-450&492_257&493_254&432_452&433-0_0_3_24_30_31_29_29-158-57.jpg
DELETED
Binary file (4.95 kB)
|
|
img/0125-89_86-273&524_472&588-466&592_269&597_270&523_467&518-0_0_18_29_27_30_32-112-29.jpg
DELETED
Binary file (983 Bytes)
|
|
img/0125-90_90-348&506_528&586-534&576_352&581_351&512_533&507-0_0_6_25_25_24_24-141-25.jpg
DELETED
Binary file (995 Bytes)
|
|
img/02-90_86-287&458_543&538-532&542_295&544_294&457_531&455-0_0_30_12_30_31_30-158-55.jpg
DELETED
Binary file (1.46 kB)
|
|
img/02-90_88-223&485_480&573-477&573_243&567_246&489_480&495-0_0_10_30_32_32_24-163-61.jpg
DELETED
Binary file (1.3 kB)
|
|
img/2022-04-08 134850.png
ADDED
img/2022-04-08 134916.png
ADDED
img/2022-04-08 134931.png
ADDED
img/2022-04-08 134950.png
ADDED
img/2022-04-08 155620.png
ADDED
plate.py
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import cv2
|
2 |
+
import matplotlib.pyplot as plt
|
3 |
+
import numpy as np
|
4 |
+
import torch
|
5 |
+
import torch.nn as nn
|
6 |
+
from torchvision import transforms
|
7 |
+
|
8 |
+
from esrgan import ESRGAN
|
9 |
+
|
10 |
+
esrgan = ESRGAN()
|
11 |
+
|
12 |
+
def viz(module, input):
|
13 |
+
x = input[0][0]
|
14 |
+
#最多显示4张图
|
15 |
+
min_num = np.minimum(4, x.size()[0])
|
16 |
+
for i in range(min_num):
|
17 |
+
plt.subplot(1, 4, i+1)
|
18 |
+
plt.imshow(x[i].cpu())
|
19 |
+
plt.xticks([]) #去掉横坐标值
|
20 |
+
plt.yticks([]) #去掉纵坐标值
|
21 |
+
plt.show()
|
22 |
+
|
23 |
+
def main():
|
24 |
+
t = transforms.Compose([transforms.ToPILImage(),
|
25 |
+
transforms.Resize((224, 224)),
|
26 |
+
transforms.ToTensor(),
|
27 |
+
transforms.Normalize(mean=[0.485, 0.456, 0.406],
|
28 |
+
std=[0.229, 0.224, 0.225])
|
29 |
+
])
|
30 |
+
|
31 |
+
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
32 |
+
|
33 |
+
model = esrgan.net
|
34 |
+
for name, m in model.named_modules():
|
35 |
+
# if not isinstance(m, torch.nn.ModuleList) and \
|
36 |
+
# not isinstance(m, torch.nn.Sequential) and \
|
37 |
+
# type(m) in torch.nn.__dict__.values():
|
38 |
+
# 这里只对卷积层的feature map进行显示
|
39 |
+
if isinstance(m, torch.nn.Conv2d):
|
40 |
+
m.register_forward_pre_hook(viz)
|
41 |
+
img = cv2.imread('image.png')
|
42 |
+
img = t(img).unsqueeze(0).to(device)
|
43 |
+
with torch.no_grad():
|
44 |
+
model(img)
|
45 |
+
|
46 |
+
if __name__ == '__main__':
|
47 |
+
main()
|
utils/__pycache__/utils.cpython-38.pyc
CHANGED
Binary files a/utils/__pycache__/utils.cpython-38.pyc and b/utils/__pycache__/utils.cpython-38.pyc differ
|
|