复制文件
#copy the file "test.py" in now path to the child path mytest/ if not os.path.exists("mytest"): os.makedirs("mytest") copy2("test.py","mytest/")
获取文件名
url="home/cx/test.txt" filename = url.split('/')[-1] print filename
利用logging输出
logging.basicConfig(format='%(asctime)s %(levelname)s %(message)s', level=logging.DEBUG, stream=sys.stdout) logging.info("Download URL: {}".format("xxxx")) logging.debug('This is debug message') logging.info('This is info message') logging.warning('This is warning message')
def get_pathes(): if 'TV_DIR_DATA' in os.environ: data_dir = os.path.join(['hypes'], os.environ['TV_DIR_DATA']) else: data_dir = "DATA" if 'TV_DIR_RUNS' in os.environ: run_dir = os.path.join(['hypes'], os.environ['TV_DIR_DATA']) else: run_dir = "RUNS" return data_dir, run_dir data_dir, run_dir = get_pathes() print os.path.join(data_dir,"cx") #if data_dir=home,then print home/cx print run_dir,data_dir
import os import logging import sys from shutil import copy2
源代码(总)
"""Download data relevant to train the KittiSeg model.""" from __future__ import absolute_import from __future__ import division from __future__ import print_function import logging import sys import os import subprocess import zipfile from six.moves import urllib from shutil import copy2 import argparse logging.basicConfig(format='%(asctime)s %(levelname)s %(message)s', level=logging.INFO, stream=sys.stdout) sys.path.insert(1, 'incl') # Please set kitti_data_url to the download link for the Kitti DATA. # # You can obtain by going to this website: # http://www.cvlibs.net/download.php?file=data_road.zip # # Replace 'http://kitti.is.tue.mpg.de/kitti/?????????.???' by the # correct URL. vgg_url = 'https://dl.dropboxusercontent.com/u/50333326/vgg16.npy' def get_pathes(): """ Get location of `data_dir` and `run_dir'. Defaut is ./DATA and ./RUNS. Alternativly they can be set by the environoment variabels 'TV_DIR_DATA' and 'TV_DIR_RUNS'. """ if 'TV_DIR_DATA' in os.environ: data_dir = os.path.join(['hypes'], os.environ['TV_DIR_DATA']) else: data_dir = "DATA" if 'TV_DIR_RUNS' in os.environ: run_dir = os.path.join(['hypes'], os.environ['TV_DIR_DATA']) else: run_dir = "RUNS" return data_dir, run_dir def download(url, dest_directory): filename = url.split('/')[-1] filepath = os.path.join(dest_directory, filename) logging.info("Download URL: {}".format(url)) logging.info("Download DIR: {}".format(dest_directory)) def _progress(count, block_size, total_size): prog = float(count * block_size) / float(total_size) * 100.0 sys.stdout.write('\r>> Downloading %s %.1f%%' % (filename, prog)) sys.stdout.flush() filepath, _ = urllib.request.urlretrieve(url, filepath, reporthook=_progress) print() return filepath def main(): parser = argparse.ArgumentParser() parser.add_argument('--kitti_url', default='', type=str) args = parser.parse_args() kitti_data_url = args.kitti_url data_dir, run_dir = get_pathes() vgg_weights = os.path.join(data_dir, 'vgg16.npy') if not os.path.exists(data_dir): os.makedirs(data_dir) # Download VGG DATA if not os.path.exists(vgg_weights): download_command = "wget {} -P {}".format(vgg_url, data_dir) logging.info("Downloading VGG weights.") download(vgg_url, data_dir) else: logging.warning("File: {} exists.".format(vgg_weights)) logging.warning("Please delete to redownload VGG weights.") data_road_zip = os.path.join(data_dir, 'data_road.zip') # Download KITTI DATA if not os.path.exists(data_road_zip): if kitti_data_url == '': logging.error("Data URL for Kitti Data not provided.") url = "http://www.cvlibs.net/download.php?file=data_road.zip" logging.error("Please visit: {}".format(url)) logging.error("and request Kitti Download link.") logging.error("Rerun scipt using" "'python download_data.py' --kitti_url [url]") exit(1) if not kitti_data_url[-19:] == 'kitti/data_road.zip': logging.error("Wrong url.") url = "http://www.cvlibs.net/download.php?file=data_road.zip" logging.error("Please visit: {}".format(url)) logging.error("and request Kitti Download link.") logging.error("Rerun scipt using" "'python download_data.py' --kitti_url [url]") exit(1) else: logging.info("Downloading Kitti Road Data.") download(kitti_data_url, data_dir) # Extract and prepare KITTI DATA logging.info("Extracting kitti_road data.") zipfile.ZipFile(data_road_zip, 'r').extractall(data_dir) kitti_road_dir = os.path.join(data_dir, 'data_road/') logging.info("Preparing kitti_road data.") train_txt = "data/train3.txt" val_txt = "data/val3.txt" testing_txt = "data/testing.txt" copy2(train_txt, kitti_road_dir) copy2(val_txt, kitti_road_dir) copy2(testing_txt, kitti_road_dir) logging.info("All data have been downloaded successful.") if __name__ == '__main__': main()