From a982d503a3b26dff0ee562d218d67786df7887a0 Mon Sep 17 00:00:00 2001 From: cellistigs Date: Tue, 31 Aug 2021 12:57:54 -0400 Subject: [PATCH] use dlc_config nj instead of calculating from data. --- src/deepgraphpose/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/deepgraphpose/dataset.py b/src/deepgraphpose/dataset.py index eea8c1c..d039a3c 100644 --- a/src/deepgraphpose/dataset.py +++ b/src/deepgraphpose/dataset.py @@ -596,7 +596,7 @@ def _compute_targets(self): dataset = create_dataset(dlc_config) nt = len(self.idxs['vis']['train']) # number of training frames # assert nt >= 1 - nj = max([dat_.joints[0].shape[0] for dat_ in dataset.data]) + nj = self.nj #Taiga edit 8/31/21: untested. max([dat_.joints[0].shape[0] for dat_ in dataset.data]) stride = dlc_config['stride'] def extract_frame_num(img_path):