Views
No views yet
1 def VGGEncoder():
2 weights = VGG16_Weights.DEFAULT
3 base_model = vgg16(weights=weights)
4 base_model.training = False
5
6 encoder_seq = nn.ModuleList()
7 moduls = nn.Sequential()
8 for layer in list(base_model.features.children()):
9 if isinstance(layer, nn.modules.pooling.MaxPool2d):
10 encoder_seq.append(moduls)
11 moduls = nn.Sequential()
12 else:
13 moduls.append(layer)
14 return encoder_seq detector = BodyDetector("model_weights/bgrm-bh.pth")1 fname = RandomSample("background folder", '*')
2 bg = LoadImage(fname)1 fname = RandomSample("image folder")
2 img = LoadImage(fname)
3 img_resize = cv2.resize(img, (224, 224), interpolation = cv2.INTER_AREA) mask = detector.DetectBody(img_resize) res = ReplaceBG(img, mask, bg)