search

vietnh1009 / Yolo-v2-pytorch

YOLO for object detection tasks
MIT License
373 stars 107 forks source link

deprecated Variable class #27

Open Sukanyasingh3 opened 11 months ago

Sukanyasingh3 commented 11 months ago

The test_coco_video.py code uses the Variable class, which is deprecated in recent versions of PyTorch. Instead, torch.tensor can be used directly.

  data = torch.tensor(image, dtype=torch.float32)