class HumanDetectionAndBlurFilterSample(object): """ human detection and blur filter sample """ def __init__(self): """ read config file and initialize auth client """ with open('./config.json', 'r') as settings: config = json.load(settings) client_id = config['CLIENT_ID'] client_secret = config['CLIENT_SECRET'] vrs_auth_client = AuthClient(client_id, client_secret) self.vrs_client = VisualRecognition(vrs_auth_client) ips_auth_client = AuthClient(client_id, client_secret) self.ips_client = ImageProcessing(ips_auth_client) def __detect_humans(self, img_path): res = self.vrs_client.detect_humans(img_path) return [human['location'] for human in res['humans']] def __blur_filter(self, img_path, options, locations): parameters = {'locations': locations, 'type': 'blur', 'options': options} res = self.ips_client.filter(img_path, parameters) return res def main(self): """ main """ parser = argparse.ArgumentParser(description='Blur filter sample') parser.add_argument('-f', '--file', type=str, dest='file_path', help='specify image file (JPEG or PNG) to process') parser.add_argument('--ksize_width', type=int, dest='ksize_width', default=31, help='specify kernel size width') parser.add_argument('--ksize_height', type=int, dest='ksize_height', default=31, help='specify kernel size height') args = parser.parse_args() if args.file_path is None: parser.print_help() return locations = self.__detect_humans(args.file_path) options = { 'locations': { 'shape': 'min_enclosing_circle', 'edge': 'blur' }, 'ksize_width': args.ksize_width, 'ksize_height': args.ksize_height, } res = self.__blur_filter(args.file_path, options, locations) if not os.path.exists('./results'): os.mkdir('./results') with open('./results/human_detection_and_blur_filter.jpg', 'wb') as f: f.write(res)
class TestMethodOK(TestCase): def setUp(self): self.aclient = Mock() self.aclient.get_access_token = Mock(return_value='atoken') self.aclient.get_api_key = Mock(return_value='apikey') self.aclient.session = Mock(return_value={'access_token': 'atoken'}) self.vrs = VisualRecognition(self.aclient) self.__hd_expected = { 'humans': [{ 'score': 1, 'location': { 'top': 100, 'right': 200, 'bottom': 200, 'left': 100 } }] } self.__fd_expected = { 'faces': [{ 'direction': 'left', 'location': { 'top': 100, 'right': 200, 'bottom': 200, 'left': 100 } }] } self.__fr_expected = { "score": 0.787753701210022, "source": { "location": { "left": 1085, "top": 244, "right": 1307, "bottom": 466 } }, "target": { "location": { "left": 659, "top": 207, "right": 812, "bottom": 360 } } } def test_param_err(self): with pytest.raises(TypeError): self.vrs.detect_faces() @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_faces_uri(self, req): req.return_value.text = json.dumps(self.__fd_expected) req.return_value.status_code = 200 assert self.__fd_expected == self.vrs.detect_faces( 'http://test.com/test.jpg') headers = make_headers('application/json') payload = json.dumps({'image': 'http://test.com/test.jpg'}) req.assert_called_once_with('POST', ENDPOINT + '/detect_faces', headers=headers, data=payload) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_faces_jpeg(self, req, opn, isfile, pil_open): req.return_value.text = json.dumps(self.__fd_expected) req.return_value.status_code = 200 opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.return_value = 'jpeg' pil_open.return_value = img assert self.__fd_expected == self.vrs.detect_faces('test.jpg') isfile.assert_called_once_with('test.jpg') pil_open.assert_called_once_with('test.jpg') opn.assert_called_once_with('test.jpg', 'rb') headers = make_headers('image/jpeg') req.assert_called_once_with('POST', ENDPOINT + '/detect_faces', headers=headers, data=opn()) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_faces_png(self, req, opn, isfile, pil_open): req.return_value.text = json.dumps(self.__fd_expected) req.return_value.status_code = 200 opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.return_value = 'png' pil_open.return_value = img assert self.__fd_expected == self.vrs.detect_faces('test.png') isfile.assert_called_once_with('test.png') pil_open.assert_called_once_with('test.png') opn.assert_called_once_with('test.png', 'rb') headers = make_headers('image/png') req.assert_called_once_with('POST', ENDPOINT + '/detect_faces', headers=headers, data=opn()) @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_humans_uri(self, req): req.return_value.text = json.dumps(self.__hd_expected) req.return_value.status_code = 200 assert self.__hd_expected == self.vrs.detect_humans( 'http://test.com/test.jpg') headers = make_headers('application/json') payload = json.dumps({'image': 'http://test.com/test.jpg'}) req.assert_called_once_with('POST', ENDPOINT + '/detect_humans', headers=headers, data=payload) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_humans_jpeg(self, req, opn, isfile, pil_open): req.return_value.text = json.dumps(self.__hd_expected) req.return_value.status_code = 200 opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.return_value = 'jpeg' pil_open.return_value = img assert self.__hd_expected == self.vrs.detect_humans('test.jpg') isfile.assert_called_once_with('test.jpg') pil_open.assert_called_once_with('test.jpg') opn.assert_called_once_with('test.jpg', 'rb') headers = make_headers('image/jpeg') req.assert_called_once_with('POST', ENDPOINT + '/detect_humans', headers=headers, data=opn()) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_humans_png(self, req, opn, isfile, pil_open): req.return_value.text = json.dumps(self.__hd_expected) req.return_value.status_code = 200 opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.return_value = 'png' pil_open.return_value = img assert self.__hd_expected == self.vrs.detect_humans('test.png') isfile.assert_called_once_with('test.png') pil_open.assert_called_once_with('test.png') opn.assert_called_once_with('test.png', 'rb') headers = make_headers('image/png') req.assert_called_once_with('POST', ENDPOINT + '/detect_humans', headers=headers, data=opn()) @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_compare_faces_uri(self, req): req.return_value.text = json.dumps(self.__fr_expected) req.return_value.status_code = 200 assert self.__fr_expected == self.vrs.compare_faces( 'http://test.com/test_1.jpg', 'http://test.com/test_2.jpg') headers = make_headers('application/json') payload = json.dumps({ 'source': 'http://test.com/test_1.jpg', 'target': 'http://test.com/test_2.jpg' }) req.assert_called_once_with('POST', ENDPOINT + '/compare_faces', headers=headers, data=payload) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_compare_faces_jpeg_jpeg(self, req, opn, isfile, pil_open): req.return_value.text = json.dumps(self.__fr_expected) req.return_value.status_code = 200 opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.side_effect = ['jpeg', 'jpeg'] pil_open.return_value = img assert self.__fr_expected == self.vrs.compare_faces( 'test_1.jpg', 'test_2.jpg') headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'} payload = {'source': opn(), 'target': opn()} req.assert_called_once_with('POST', ENDPOINT + '/compare_faces', headers=headers, files=payload) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_compare_faces_png_png(self, req, opn, isfile, pil_open): req.return_value.text = json.dumps(self.__fr_expected) req.return_value.status_code = 200 opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.side_effect = ['png', 'png'] pil_open.return_value = img assert self.__fr_expected == self.vrs.compare_faces( 'test_1.png', 'test_2.png') headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'} payload = {'source': opn(), 'target': opn()} req.assert_called_once_with('POST', ENDPOINT + '/compare_faces', headers=headers, files=payload) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_compare_faces_jpeg_png(self, req, opn, isfile, pil_open): req.return_value.text = json.dumps(self.__fr_expected) req.return_value.status_code = 200 opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.side_effect = ['jpeg', 'png'] pil_open.return_value = img assert self.__fr_expected == self.vrs.compare_faces( 'test_1.jpeg', 'test_2.png') headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'} payload = {'source': opn(), 'target': opn()} req.assert_called_once_with('POST', ENDPOINT + '/compare_faces', headers=headers, files=payload)
class TestMethodError(TestCase): def setUp(self): self.aclient = Mock() self.aclient.get_access_token = Mock(return_value='atoken') self.aclient.get_api_key = Mock(return_value='apikey') self.aclient.session = Mock(return_value={'access_token': 'atoken'}) self.vrs = VisualRecognition(self.aclient) def test_detect_humans_file_not_found(self): with pytest.raises(ValueError) as excinfo: self.vrs.detect_humans('image.jpg') assert util.RESOURCE_ERROR == str(excinfo.value) def test_detect_faces_file_not_found(self): with pytest.raises(ValueError) as excinfo: self.vrs.detect_faces('image.jpg') assert util.RESOURCE_ERROR == str(excinfo.value) def test_compare_faces_file_not_found(self): with pytest.raises(ValueError) as excinfo: self.vrs.compare_faces('image_1.jpg', 'image_2.jpg') assert util.RESOURCE_ERROR == str(excinfo.value) @mock.patch('json.loads') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_humans_resource_not_found(self, req, ret): req.return_value.text = 'test' req.return_value.status_code = 404 ret.return_value = { 'error': { 'code': 'resource_not_found', 'message': 'The specified resource does not exist.' } } try: self.vrs.detect_humans('http://test.com/test.jpg') except ClientError as excinfo: assert ret.return_value == excinfo.response assert req.return_value.status_code == excinfo.status_code @mock.patch('json.loads') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_faces_resource_not_found(self, req, ret): req.return_value.text = 'test' req.return_value.status_code = 404 ret.return_value = { 'error': { 'code': 'resource_not_found', 'message': 'The specified resource does not exist.' } } try: self.vrs.detect_faces('http://test.com/test.jpg') except ClientError as excinfo: assert ret.return_value == excinfo.response assert req.return_value.status_code == excinfo.status_code @mock.patch('json.loads') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_compare_faces_resource_not_found(self, req, ret): req.return_value.text = 'test' req.return_value.status_code = 404 ret.return_value = { 'error': { 'code': 'resource_not_found', 'message': 'The specified resource does not exist.' } } try: self.vrs.compare_faces('http://test.com/test_1.jpg', 'http://test.com/test_2.jpg') except ClientError as excinfo: assert ret.return_value == excinfo.response assert req.return_value.status_code == excinfo.status_code @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_humans_gif(self, req, opn, isfile, pil_open): req.side_effect = RequestException opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.return_value = 'gif' pil_open.return_value = img with pytest.raises(ValueError) as excinfo: self.vrs.detect_humans('image.gif') assert util.UNSUPPORTED_ERROR == str(excinfo.value) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_detect_faces_gif(self, req, opn, isfile, pil_open): req.side_effect = RequestException opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.return_value = 'gif' pil_open.return_value = img with pytest.raises(ValueError) as excinfo: self.vrs.detect_faces('image.gif') assert util.UNSUPPORTED_ERROR == str(excinfo.value) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') @mock.patch('ricohcloudsdk.vrs.util.SESSION.request') def test_compare_faces_gif(self, req, opn, isfile, pil_open): req.side_effect = RequestException opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.return_value = True img = MagicMock() img.format.lower.return_value = 'gif' pil_open.return_value = img with pytest.raises(ValueError) as excinfo: self.vrs.compare_faces('image_1.gif', 'image_2.gif') assert util.UNSUPPORTED_ERROR == str(excinfo.value) @mock.patch('ricohcloudsdk.vrs.util.Image.open') @mock.patch('os.path.isfile') @mock.patch('ricohcloudsdk.vrs.client.open') def test_compare_faces_jpeg_uri(self, opn, isfile, pil_open): opn.side_effect = mock.mock_open() opn.read_data = b'readdata' isfile.side_effect = [True, False] img = MagicMock() img.format.lower.return_value = 'jpeg' pil_open.return_value = img with pytest.raises(ValueError) as excinfo: self.vrs.compare_faces('test_1.jpeg', 'https://test.co,/test.jpg') assert util.COMBINATION_ERROR == str(excinfo.value)