Commit 1d908417 authored by dzy7e's avatar dzy7e
Browse files

TEST_TRANSFORM no CenterCrop

parent fef3de6d

zoo/ccip/dataset.py

+2 −2
Original line number Diff line number Diff line
@@ -20,8 +20,8 @@ TRAIN_TRANSFORM = [ 
    transforms.ToTensor(),

]

TEST_TRANSFORM = [

    transforms.Resize((416, 416)),

    transforms.CenterCrop(384),

    transforms.Resize((384, 384)),

    #transforms.c(384),

    transforms.ToTensor(),

]