diff --git a/README.md b/README.md
index dac0a20b3d2c09f1967078c08e7893fd01542bb6..01d08342a750e3ed45b26badb2ac148e4b258de5 100755
--- a/README.md
+++ b/README.md
@@ -3,14 +3,16 @@
You can use it to automatically remove the mosaics in images and videos, or add mosaics to them.
This porject based on semantic segmentation and pix2pix.
-## Notes
+
+## Notice
The code do not include the part of training, I will finish it in my free time.
+
## Prerequisites
-- Linux, (I didn't try this code on Windows or mac machine)
-- Python 3.5+
+- Linux, (I didn't try this code on Windows or Mac OS)
+- Python 3.6+
- ffmpeg
-- Pytroch 0.4, (I will update to 1.0)
+- Pytorch 1.0+ [(Old version codes)](https://github.com/HypoX64/DeepMosaics/tree/Pytorch0.4)
- CPU or NVIDIA GPU + CUDA CuDNN
## Getting Started
@@ -21,8 +23,9 @@ cd DeepMosaics
```
### Get pre_trained models and test video
You can download pre_trained models and test video and replace the files in the project.
-[[Google Drive]](https://drive.google.com/open?id=1PXt3dE9Eez2xUqpemLJutwTCC0tW-D2g)
- [[百度云,提取码z8vz]](https://pan.baidu.com/s/1Wi8T6PE4ExTjrHVQhv3rJA)
+[[Google Drive]](https://drive.google.com/open?id=10nARsiZoZGcaKw40nQu9fJuRp1oeabPs)
+ [[百度云,提取码7thu]](https://pan.baidu.com/s/1IG4bdIiIC9PH9-oEyae5Sg)
+
### Dependencies
This code depends on numpy, scipy, opencv-python, torchvision, available via pip install.
### AddMosaic