diff --git a/requirements.txt b/requirements.txt index 0f6fc4e..473d345 100644 --- a/requirements.txt +++ b/requirements.txt @@ -28,6 +28,7 @@ pytorch_msssim pytorch-lightning # for process images & videos +rembg imageio imageio-ffmpeg matplotlib