Update README.md

README.md

@@ -5,6 +5,7 @@ tags:
 - image-segmentation
 - historical-maps
 - cartography
+library: pytorch
 ---
 
 # Semantic Segmentation in Historical Maps
@@ -36,4 +37,4 @@ python inference.py \
   --out-dir predictions/
 ```
 
-`inference.py` is a minimal example: sliding-window prediction with `stride = patch_size // 2`, ImageNet normalization, sigmoid + per-class thresholding, writing one binary PNG per class to `--out-dir`. For the full training and inference pipeline see the [GitHub repository](https://github.com/dav1dclara/cartography-research).
+`inference.py` is a minimal example: sliding-window prediction with `stride = patch_size // 2`, ImageNet normalization, sigmoid + per-class thresholding, writing one binary PNG per class to `--out-dir`. For the full training and inference pipeline see the [GitHub repository](https://github.com/dav1dclara/cartography-research).