opensource/DeOldify
3
0
Fork 0
mirror of https://github.com/jantic/DeOldify.git synced 2026-08-30 18:02:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Attempting to make image url logic more robust

Browse Source
This commit is contained in:
Jason Antic
2019-04-16 17:55:11 -07:00
parent 57941117f8
commit ea57ecf75a
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ImageColorizerColab.ipynb
+2 -2
View File
@@ -225,7 +225,7 @@
"outputs": [],
"source": [
"if source_url is not None and source_url !='':\n",
" colorizer.plot_transformed_image_from_url(path=\"test_images/image.jpg\", url=source_url, render_factor=render_factor)\n",
" colorizer.plot_transformed_image_from_url(url=source_url, render_factor=render_factor)\n",
"else:\n",
" print('Provide an image url and try again.')"
]
@@ -241,7 +241,7 @@
"\n",
"* In the menu to the left, click **Files**\n",
"* If you don't see the 'DeOldify' folder, click \"Refresh\"\n",
"* By default, rendered image will be in /DeOldify/result_images/"
"* By default, rendered image will be in /DeOldify/result_images/image.png"
]
},
{
fasterai/visualize.py
+2 -2
View File
@@ -30,9 +30,9 @@ class ModelImageVisualizer():
def _open_pil_image(self, path:Path)->Image:
return PIL.Image.open(path).convert('RGB')
def plot_transformed_image_from_url(self, path:str, url:str, figsize:(int,int)=(20,20), render_factor:int=None)->Image:
def plot_transformed_image_from_url(self, url:str, path:str='test_images/image.png', figsize:(int,int)=(20,20), render_factor:int=None)->Image:
response = requests.get(url)
img = Image.open(BytesIO(response.content))
img = Image.open(BytesIO(response.content)).convert('RGB')
img.save(path)
return self.plot_transformed_image(path=path, figsize=figsize, render_factor=render_factor)