Update README.md
Browse files
README.md
CHANGED
@@ -33,4 +33,18 @@ model_index:
|
|
33 |
|
34 |
PercepNet model trained on TID2008 and validated on TID2013, obtaining 0.97 and 0.93 Pearson Correlation respectively.
|
35 |
|
36 |
-
Link to the run: https://wandb.ai/jorgvt/PerceptNet/runs/28m2cnzj?workspace=user-jorgvt
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
|
34 |
PercepNet model trained on TID2008 and validated on TID2013, obtaining 0.97 and 0.93 Pearson Correlation respectively.
|
35 |
|
36 |
+
Link to the run: https://wandb.ai/jorgvt/PerceptNet/runs/28m2cnzj?workspace=user-jorgvt
|
37 |
+
|
38 |
+
# Usage
|
39 |
+
|
40 |
+
As of now to use the model you have to install the [PerceptNet repo](https://github.com/Jorgvt/perceptnet) to get access to the `PerceptNet` class where you will load the weights available here like this:
|
41 |
+
|
42 |
+
```python
|
43 |
+
from perceptnet.networks import PerceptNet
|
44 |
+
|
45 |
+
weights_path = get_file(fname='perceptnet_rgb.h5',
|
46 |
+
origin='https://huggingface.co/Jorgvt/PerceptNet/blob/main/final_model_rgb.h5')
|
47 |
+
model = PerceptNet(kernel_initializer='ones', gdn_kernel_size=1, learnable_undersampling=False)
|
48 |
+
model.build(input_shape=(None, 384, 512, 3))
|
49 |
+
model.load_weights(weights_path)
|
50 |
+
```
|