Spaces:

piyushgrover
/

yolov3

Sleeping

piyushgrover commited on Aug 14, 2023

Commit

6d42992

1 Parent(s): d1be1c8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ device = torch.device('cpu')
 dataset_mean, dataset_std = (0.4914, 0.4822, 0.4465), \
                             (0.2470, 0.2435, 0.2616)
 model = YoloV3Lightning.YOLOv3LightningModel(num_classes=cfg.NUM_CLASSES, anchors=cfg.ANCHORS, S=cfg.S)
-ckpt_file = 'ckpt_light.pth'
 checkpoint = load_model_from_checkpoint(device, file_name=ckpt_file)
 model.load_state_dict(checkpoint['model'], strict=False)

 dataset_mean, dataset_std = (0.4914, 0.4822, 0.4465), \
                             (0.2470, 0.2435, 0.2616)
 model = YoloV3Lightning.YOLOv3LightningModel(num_classes=cfg.NUM_CLASSES, anchors=cfg.ANCHORS, S=cfg.S)
+ckpt_file = 'ckpt_light2.pth'
 checkpoint = load_model_from_checkpoint(device, file_name=ckpt_file)
 model.load_state_dict(checkpoint['model'], strict=False)