add tensorboardX visualize

Author: naviocean

.idea/workspace.xml

@@ -22,6 +22,7 @@
 parser.add_argument('--pretrained', dest='pretrained', action='store_true', help='use pre-trained model')
 parser.add_argument("--model-type", default="P3D", choices=["P3D", "C3D", "I3D"], help="which model to run the code")
 parser.add_argument('--num-frames', default=16, type=int, metavar='N', help='number frames per clip')
+parser.add_argument('--log-visualize', default='./runs', type=str, metavar='PATH', help='tensorboard log')
 
 def main():
     args = parser.parse_args()

main.py

@@ -102,8 +102,7 @@ def loading_data(self):
         normalize = Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
 
         val_transformations = Compose([
-            Resize((182, 242)),
-            CenterCrop(size),
+            Resize((size, size)),
             ToTensor(),
             normalize
         ])