Open Camera or PhotoLibrary with only images // swift 3

//// Vladimir Zhelnov - neatek.pw - Web/iOS dev
// +
import MobileCoreServices
// add into Classes -> UIImagePickerControllerDelegate

let picker = UIImagePickerController()

override func viewDidLoad() {
        picker.delegate = self
}

// Open gallery with only images
@IBAction func go_photo(_ sender: AnyObject) {
            if UIImagePickerController.isSourceTypeAvailable(UIImagePickerControllerSourceType.photoLibrary) {
                self.picker.allowsEditing = false
                self.picker.sourceType = UIImagePickerControllerSourceType.photoLibrary
                self.picker.mediaTypes = [kUTTypeImage as String] // need import Mobile..
                //self.picker.mediaTypes = UIImagePickerController.availableMediaTypes(for: .photoLibrary)!
                self.present(self.picker, animated: true, completion: nil)
            }
}

// Open Camera
@IBAction func go_photo_2(_ sender: AnyObject) {
            if UIImagePickerController.isSourceTypeAvailable(UIImagePickerControllerSourceType.camera) {
                //var imagePicker = UIImagePickerController()
                self.picker.delegate = self
                self.picker.sourceType = UIImagePickerControllerSourceType.camera;
                self.picker.allowsEditing = false
                self.present(self.picker, animated: true, completion: nil)
            }
}