From 179ed1d41dcde593a3cd812962ab7ed25412632d Mon Sep 17 00:00:00 2001 From: WANG KAIXUAN Date: Thu, 25 Apr 2024 21:15:48 +0800 Subject: [PATCH] Update track_predictor_cotracker.py fix torch.hub download HTTP Error 403 --- flowmap/tracking/track_predictor_cotracker.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/flowmap/tracking/track_predictor_cotracker.py b/flowmap/tracking/track_predictor_cotracker.py index fefe789..51f6893 100644 --- a/flowmap/tracking/track_predictor_cotracker.py +++ b/flowmap/tracking/track_predictor_cotracker.py @@ -20,6 +20,9 @@ class TrackPredictorCoTrackerCfg: class TrackPredictorCoTracker(TrackPredictor[TrackPredictorCoTrackerCfg]): def __init__(self, cfg: TrackPredictorCoTrackerCfg) -> None: super().__init__(cfg) + # fix torch.hub HTTP Error 403 + # according to https://github.com/pytorch/vision/issues/4156#issuecomment-886005117 + torch.hub._validate_not_a_forked_repo=lambda a,b,c: True self.tracker = torch.hub.load( "facebookresearch/co-tracker:v1.0", "cotracker_w8" )