pd_repo_test.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"name":"pd_repo_test.ipynb","provenance":[],"collapsed_sections":[],"authorship_tag":"ABX9TyM1wjv3u29CdfosLsKlzvXE"},"kernelspec":{"name":"python3","display_name":"Python 3"},"language_info":{"name":"python"}},"cells":[{"cell_type":"code","metadata":{"colab":{"base_uri":"https://localhost:8080/"},"id":"zkM2QvIMwhuy","executionInfo":{"status":"ok","timestamp":1627302030671,"user_tz":-60,"elapsed":274,"user":{"displayName":"Martin Marek","photoUrl":"https://lh3.googleusercontent.com/a-/AOh14GhPMe9bH4uZG5V-exmgiiZJtm58T0EKUuVCQ0ONxck=s64","userId":"04932572550491068578"}},"outputId":"a0f2cd7d-3a19-4450-bcd3-44b0d1e2e1d8"},"source":["import os\n","! git clone https://github.com/martin-marek/parking-space-occupancy\n","os.chdir('parking-space-occupancy')"],"execution_count":2,"outputs":[{"output_type":"stream","text":["fatal: destination path 'parking-space-occupancy' already exists and is not an empty directory.\n"],"name":"stdout"}]},{"cell_type":"code","metadata":{"colab":{"base_uri":"https://localhost:8080/"},"id":"0Ijuda30wiLp","executionInfo":{"status":"ok","timestamp":1627302078395,"user_tz":-60,"elapsed":1395,"user":{"displayName":"Martin Marek","photoUrl":"https://lh3.googleusercontent.com/a-/AOh14GhPMe9bH4uZG5V-exmgiiZJtm58T0EKUuVCQ0ONxck=s64","userId":"04932572550491068578"}},"outputId":"a4af46ff-059d-4a05-94a8-3ea8d6457107"},"source":["import torch, os, requests\n","from models.rcnn import RCNN\n","from utils import transforms\n","\n","# create model\n","model = RCNN()\n","\n","# load model weights\n","weights_path = 'weights.pt'\n","if not os.path.exists(weights_path):\n","    r = requests.get('https://storage.googleapis.com/pd-models/RCNN_128_square_gopro.pt')  \n","    with open(weights_path, 'wb') as f:\n","        f.write(r.content)\n","model.load_state_dict(torch.load(weights_path, map_location='cpu'))\n","\n","# inference\n","image = torch.zeros([3, 1000, 1000])\n","rois = torch.zeros([10, 4, 2])\n","image = transforms.preprocess(image)\n","class_logits = model(image, rois)\n","class_scores = class_logits.softmax(1)[:, 1]"],"execution_count":6,"outputs":[{"output_type":"execute_result","data":{"text/plain":["torch.Size([10])"]},"metadata":{"tags":[]},"execution_count":6}]},{"cell_type":"code","metadata":{"id":"wff4vo5qwnX6"},"source":[""],"execution_count":null,"outputs":[]}]}