28 Commits

Author SHA1 Message Date
Jason Antic 4ff02e5f22 Fixing RGB/BGR conversion issue
Boy was this silly...
2023-03-29 11:13:03 -04:00
Gareth Davidson dff82b6d2e Remove execution bit
All the files marked as executable were bugging me, so I ran this:

find . -executable -type f -print0 | xargs -0 grep -L '#!' | xargs chmod -x
2022-09-19 08:50:00 +01:00
Jason Antic 8a3f56b028 Improving on ffmpeg error reporting and getting rid of extraneous ffmpeg output; Using logging instead of print 2022-07-18 18:39:25 -07:00
Jason Antic d5a716bd5a Getting rid of Docker support; youtube_dl -> yt_dlp; fixing installation 2022-07-08 20:14:25 -07:00
Jason Antic 2a003d999d Fixing issue downloading certain images by url (from Imgur, for example) 2022-01-10 10:11:05 -08:00
Jason Antic 6422160a0e Fixing sloppy imports and syntax errors 2021-12-13 09:33:57 -08:00
Jason Antic 17fbe6b950 Minor cpu performance fix and typo fix. 2021-05-27 16:48:07 -07:00
Jason Antic ed66f69a71 Minor cleanup to results_dir functionality that was recently merged 2021-01-12 16:18:00 -08:00
Mike 8320bee69b Tweaks to make it more convenient to call Deoldify app from an external python script that may not want to use Deoldify's expected working directory.
- Give get_image_colorizer() ability to receive root_folder as argument.

- Give plot_transformed_image() ability to receive results directory as an argument.
2020-12-31 06:26:10 -08:00
Uesley Carossi 26fa0fd46c Add accept header for permit all MIME types 2020-11-12 11:30:22 -03:00
Jason Antic adb23e742e Bumping up retries on youtube_dl 2020-07-30 16:43:05 -07:00
Jason Antic 7f74755818 Adding retries to make video downloads more reliable. Adding disclaimer about longer videos on Colab 2020-07-30 16:36:00 -07:00
Jason Antic ee1f8d2f3b Adding proper attribution 2020-06-18 19:17:11 -07:00
Jason Antic 696001361a Adding graceful and informative gpu out of memory error handling. 2020-04-21 18:08:45 -07:00
Jason Antic e55c1abc86 Making switching between various GPUS and CPU easy. 2020-03-27 18:21:42 -07:00
Jason Antic 4969dfa11d Fixing a few bugs and cleaning up new post_process/stats parameterizations 2020-03-26 16:24:05 -07:00
AIEMMU 0d129fc9ed add ability to add your own stats not just imagenet_stats to the dataloader and filter. Also allowing the user to get the image before it is preprocessed. 2020-03-26 10:26:11 +00:00
Jason Antic 51991de5e6 oops 2020-02-20 09:04:15 -08:00
Jason Antic 48efb2c46e Undoing watermark changes as these too apparently broke on Colab (wat) 2020-02-20 08:58:19 -08:00
Jason Antic b1c45227bd Undoing plt close logic as it appears to be breaking Colabs (doh!) 2020-02-20 08:53:50 -08:00
Jason Antic 227548d8ad Improving memory management of plt; Loading watermark only once. 2020-02-19 19:03:17 -08:00
Jason Antic 1edfcc6253 Fixing bug in watermark logic 2020-02-18 18:58:23 -08:00
Jason Antic 2326e4090b Adding new palette watermark 2020-02-18 17:35:05 -08:00
Alexandre Vicenzi 6c6e2ee6e4 Remove default loss function from generators
If the loss function is set as a default argument it will be loaded even
if not used, which is not desired.

FeatureLoss is used only for training, but was being loaded also for
inference, using a little more memory than required due VGG16 features.

This will not cause any harm because training and inference usage always
set gen_loss argument.
2019-10-28 23:27:51 +01:00
Alexandre Vicenzi 3fb4956fdd Refactor and cleanup
Black was used to format the code as well some pylint issues where
fixed.
2019-10-19 14:31:49 +02:00
Alexandre Vicenzi 2346eb07b5 Add timeout for image download
If the image download get stuck
requests.get() will hang forever.

This patch includes a timeout of
30s to avoid this behavior.

This fixes #151.
2019-08-23 21:38:05 +02:00
Alexandre Vicenzi b80aaafe09 code cleanup and refactor
Using Black and PyLint to improve
code quality.
2019-08-22 21:10:01 +02:00
Alexandre Vicenzi 547fb7e56f pypi: rename fasterai to deoldify
It will be better to do:

    from deoldify import visualize

Than:

    from fasterai import visualize

The PyPI package will be called DeOldify, in this case
it makes more sense to have an import name that matches
the package name.

Also, fasterai resembles fastai library, and DeOldify
is not a library to use with fastai, it's built on
top of it.
2019-08-20 21:22:00 +02:00