diff --git a/i2v/caffe_i2v.py b/i2v/caffe_i2v.py index 63ee5e2c..9e32cc2d 100644 --- a/i2v/caffe_i2v.py +++ b/i2v/caffe_i2v.py @@ -21,6 +21,7 @@ def _extract(self, inputs, layername): -self.net.crop_dims / 2.0, self.net.crop_dims / 2.0 ]) + crop = [int(i) for i in crop] input_ = input_[:, crop[0]:crop[2], crop[1]:crop[3], :] # Classify caffe_in = np.zeros(