python-3.x - 在这种情况下看门狗的 on_created 的事件参数是什么?
问题描述
我希望我的看门狗类中的 on_created 方法返回一个值。但是,我不知道 on_created 函数的事件参数是什么,所以我可以调用它。
这是我的看门狗类和 on_created 函数
class FrameHandler(FileSystemEventHandler):
def on_created(self, event): # when file is created
# Produce evaluation for the frame
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
path = '/home/Desktop/mask-detector-model.pt'
model_ft = models.mobilenet_v2(pretrained=True)
model_ft.classifier[1] = torch.nn.Linear(in_features=model_ft.classifier[1].in_features, out_features=3)
model_ft.load_state_dict(torch.load(path, map_location = device))
class_names = ['MASKED', 'NOT_MASKED', 'NO_PERSON']
image_transforms = transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize([0.5237], [0.2611]) ])
data_dir = '/home/Desktop/Frames'
test_dataset = datasets.ImageFolder(root=data_dir, transform= image_transforms)
test_loader = torch.utils.data.DataLoader(test_dataset, batch_size=1, shuffle=True, num_workers=1, pin_memory=True)
with torch.no_grad():
for i, (inputs, labels) in enumerate(test_loader):
outputs = model_ft(inputs)
_, preds = torch.max(outputs, 1)
for j in range(inputs.size()[0]):
prediction = class_names[preds[j]]
return prediction
解决方案
推荐阅读
- powershell - 在 PowerShell 中创建新的 Outlook 365 电子邮件
- javascript - 如何通过不在引号中的正则表达式拆分文本
- python - 如何在python中编辑矩阵中的值?
- ascii - 为什么 '[' 和 ']' ASCII 码不相互跟随?
- python - OSError: [WinError 126] 未找到模块
- javascript - Github 网页外部 css 无法正常工作,但 js 工作正常
- docker - Nuxt.js SSR w/ Nest API 在 Docker 容器中部署到 AWS
- android - 如何在flutter中使用fxpoi包读取CSV文件
- python - 使用 keras 将图像划分为子部分
- python - Python同时监听和分析数据