diff --git a/ML/Pytorch/GANs/Pix2Pix/README.md b/ML/Pytorch/GANs/Pix2Pix/README.md index b6297a3..5957572 100644 --- a/ML/Pytorch/GANs/Pix2Pix/README.md +++ b/ML/Pytorch/GANs/Pix2Pix/README.md @@ -17,13 +17,15 @@ The dataset can be downloaded from Kaggle: [link](https://www.kaggle.com/vikramt The dataset can be downloaded from Kaggle: [link](https://www.kaggle.com/ktaebum/anime-sketch-colorization-pair). ### Download pretrained weights -Pretrained weights for Satellite image to Google Map [here](https://github.com/aladdinpersson/Machine-Learning-Collection/releases/download/1.0/Pix2Pix_Weights_Satellite_to_Map.zip) -Pretrained weights for Colorizing Anime [here](https://github.com/aladdinpersson/Machine-Learning-Collection/releases/download/1.0/Pix2Pix_Weights_Colorize_Anime.zip) +Pretrained weights for Satellite image to Google Map [here](https://github.com/aladdinpersson/Machine-Learning-Collection/releases/download/1.0/Pix2Pix_Weights_Satellite_to_Map.zip). + +Pretrained weights for Colorizing Anime [here](https://github.com/aladdinpersson/Machine-Learning-Collection/releases/download/1.0/Pix2Pix_Weights_Colorize_Anime.zip). + +Extract the zip file and put the pth.tar files in the directory with all the python files. Make sure you put LOAD_MODEL=True in the config.py file. ### Training Edit the config.py file to match the setup you want to use. Then run train.py - ## Pix2Pix paper ### Image-to-Image Translation with Conditional Adversarial Networks by Phillip Isola, Jun-Yan Zhu, Tinghui Zhou, Alexei A. Efros diff --git a/README.md b/README.md index c18828a..a5573ce 100644 --- a/README.md +++ b/README.md @@ -71,14 +71,18 @@ If you have any specific video suggestion please make a comment on YouTube :) * [![Youtube Link][logo]](https://youtu.be/XXYG5ZWtjj0)   [Intersection over Union](https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/object_detection/metrics/iou.py) * [![Youtube Link][logo]](https://youtu.be/YDkjWEN8jNA)   [Non-Max Suppression](https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/object_detection/metrics/nms.py) * [![Youtube Link][logo]](https://youtu.be/FppOzcDvaDI)   [Mean Average Precision](https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/object_detection/metrics/mean_avg_precision.py) -* [![Youtube Link][logo]](https://youtu.be/n9_XyCGr-MI)   [YOLO from scratch](https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/object_detection/YOLO) +* [![Youtube Link][logo]](https://youtu.be/n9_XyCGr-MI)   [YOLOv1 from scratch](https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/object_detection/YOLO) +* [![Youtube Link][logo]](https://youtu.be/Grir6TZbc1M)   [YOLOv3 from scratch](https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/Pytorch/object_detection/YOLOv3) ### Generative Adversarial Networks * [![Youtube Link][logo]](https://youtu.be/OljTVUVzPpM)   [Simple FC GAN](https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/GANs/1.%20SimpleGAN/fc_gan.py) * [![Youtube Link][logo]](https://youtu.be/IZtv9s_Wx9I)   [DCGAN](https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/Pytorch/GANs/2.%20DCGAN) * [![Youtube Link][logo]](https://youtu.be/pG0QZ7OddX4)   [WGAN](https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/Pytorch/GANs/3.%20WGAN) * [![Youtube Link][logo]](https://youtu.be/pG0QZ7OddX4)   [WGAN-GP](https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/Pytorch/GANs/4.%20WGAN-GP) -* [ProGAN (video coming soon)](https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/Pytorch/GANs/5.%20ProGAN) +* [![Youtube Link][logo]](https://youtu.be/SuddDSqGRzg)   [Pix2Pix](https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/Pytorch/GANs/4.%20WGAN-GP) +* [![Youtube Link][logo]]()   [CycleGAN (coming soon)]() +* [![Youtube Link][logo]]()   [SRGAN(coming soon)]() +* [![Youtube Link][logo]]()   [ProGAN (coming soon)]() ### Architectures * [![Youtube Link][logo]](https://youtu.be/fcOW-Zyb5Bo)   [LeNet5](https://github.com/AladdinPerzon/Machine-Learning-Collection/blob/79f2e1928906f3cccbae6c024f3f79fd05262cd1/ML/Pytorch/CNN_architectures/lenet5_pytorch.py#L15-L35) **- CNN architecture**