update codes

lib/utils/__init__.py

@@ -4,7 +4,6 @@ from .utils import test_imagenet_data
 from .utils import print_log
 from .evaluation_utils import obtain_accuracy
 from .draw_pts import draw_points
-from .fb_transform import ApplyOffset
 from .gpu_manager import GPUManager
 
 from .save_meta import Save_Meta

lib/utils/fb_transform.py

@@ -1,14 +0,0 @@
-import torch
-import random
-import numpy as np
-
-class ApplyOffset(object):
-  def __init__(self, offset):
-    assert isinstance(offset, int), 'The offset is not right : {}'.format(offset)
-    self.offset = offset
-  def __call__(self, x):
-    if isinstance(x, np.ndarray) and x.dtype == 'uint8':
-      x = x.astype(int)
-    if isinstance(x, np.ndarray) and x.size == 1:
-      x = int(x)
-    return x + self.offset