Ensure integer division of img size

Author: jmiller656

test.py

@@ -19,7 +19,7 @@
 if not os.path.exists(args.outdir):
 	os.mkdir(args.outdir)
 data.load_dataset(args.dataset)
-down_size = args.imgsize/args.scale
+down_size = args.imgsize//args.scale
 network = EDSR(down_size,args.layers,args.featuresize,scale=args.scale)
 network.resume(args.savedir)
 if args.image:

train.py

@@ -12,7 +12,7 @@
 parser.add_argument("--iterations",default=1000,type=int)
 args = parser.parse_args()
 data.load_dataset(args.dataset)
-down_size = args.imgsize/args.scale
+down_size = args.imgsize//args.scale
 network = EDSR(down_size,args.layers,args.featuresize,args.scale)
 network.set_data_fn(data.get_batch,(args.batchsize,args.imgsize,down_size),data.get_test_set,(args.imgsize,down_size))
 network.train(args.iterations,args.savedir)