Download converted PyTorch model weights

models/get-models.sh

@@ -48,14 +48,19 @@ if [ ! -f dlib/mmod_human_face_detector.dat ]; then
 fi
 
 mkdir -p openface
-if [ ! -f openface/nn4.small2.v1.t7 ]; then
+if [ ! -f openface/nn4.small2.v1.pt ]; then
   printf "\n\n====================================================\n"
   printf "Downloading OpenFace models, which are copyright\n"
   printf "Carnegie Mellon University and are licensed under\n"
   printf "the Apache 2.0 License.\n\n"
-  printf "This will incur about 100MB of network traffic for the models.\n"
+  printf "This will incur about 50MB of network traffic for the models.\n"
   printf "====================================================\n\n"
 
+  wget -nv \
+       https://storage.cmusatyalab.org/openface-models/nn4.small2.v1.pt \
+       -O openface/nn4.small2.v1.pt
+  [ $? -eq 0 ] || ( rm openface/nn4.small2.v1.pt* && die "+ nn4.small2.v1.pt: Error in wget." )
+
   wget -nv \
        https://storage.cmusatyalab.org/openface-models/nn4.small2.v1.t7 \
        -O openface/nn4.small2.v1.t7
@@ -117,3 +122,7 @@ checkmd5 \
 checkmd5 \
   openface/nn4.small2.v1.t7 \
   c95bfd8cc1adf05210e979ff623013b6
+
+checkmd5 \
+  openface/nn4.small2.v1.pt \
+  8de23b5e35e49df171175d28847c67c4