Skip to content
Snippets Groups Projects
Commit f2356e08 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

small fixes

parent e730e352
Branches
Tags
1 merge request!18MCCNN trainer
import numpy
import numpy as np
import torch
from torch.autograd import Variable
......
......@@ -322,7 +322,7 @@ def test_extractors():
assert output.shape[0] == 256
# MCCNN
from . import MCCNNExtractor
from ..extractor.image import MCCNNExtractor
extractor = MCCNNExtractor(num_channels=4)
# this architecture expects num_channelsx128x128 Multi channel images
data = numpy.random.rand(4, 128, 128).astype("float32")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment