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 |
+
```
|