def get_model(gpu):
model = FasterRCNN(gpu)
model.train = False
serializers.load_npz('data/VGG16_faster_rcnn_final.model', model)
return model
微信
公众号
扫码关注公众号