straighten-img: file from args

1 files changed, 2 insertions, 2 deletions

diff --git a/straighten-img.cpp b/straighten-img.cpp
index 102e8e5..2b83b85 100644
--- a/straighten-img.cpp
+++ b/straighten-img.cpp
@@ -65,9 +65,9 @@ void on_mouse(int event, int x, int y, int flags, void* param)
 		cv::imshow("src", im1);
 }
 
-int main()
+int main(int argc, char** argv)
 {
-	im0 = cv::imread("lena.png");
+	im0 = cv::imread(argv[1]);
 	if (!im0.data)
 	{
 		std::cout << "Cannot load source image!" << std::endl;