Add cv2 to the requirements

Add loading ckpt logic from state_dict and not directly from model class

Add wandb to requirements

Setup