Ashrafb commited on
Commit
e530dd3
·
verified ·
1 Parent(s): f4dd637

Update vtoonify_model.py

Browse files
Files changed (1) hide show
  1. vtoonify_model.py +1 -5
vtoonify_model.py CHANGED
@@ -223,8 +223,4 @@ class Model():
223
  except Exception as e:
224
  logging.error(f"Error during model execution: {e}")
225
  return np.zeros((256, 256, 3), np.uint8), f"Error during processing: {str(e)}"
226
- def tensor2cv2(self, img):
227
- """Convert a tensor image to OpenCV format."""
228
- tmp = ((img.cpu().numpy().transpose(1, 2, 0) + 1.0) * 127.5).astype(np.uint8).copy()
229
- logging.debug(f"Converted image shape: {tmp.shape}, strides: {tmp.strides}")
230
- return cv2.cvtColor(tmp, cv2.COLOR_RGB2BGR)
 
223
  except Exception as e:
224
  logging.error(f"Error during model execution: {e}")
225
  return np.zeros((256, 256, 3), np.uint8), f"Error during processing: {str(e)}"
226
+