[~] Refactor

This commit is contained in:
Siarhei Siniak 2021-07-26 13:46:32 +03:00
parent b6697e71b1
commit ad79a243d3
4 changed files with 29 additions and 1 deletions

3
.gitmodules vendored Normal file

@ -0,0 +1,3 @@
[submodule "deps/alpha-pose"]
path = deps/alpha-pose
url = https://github.com/WildflowerSchools/AlphaPose

@ -50,5 +50,5 @@ for o1, o2 in [
assert os.system(r''' assert os.system(r'''
mkdir -p /kaggle/working/test-input && mkdir -p /kaggle/working/test-output && cp /kaggle/working/AlphaPose/examples/demo/*.jpg /kaggle/working/test-input mkdir -p /kaggle/working/test-input && mkdir -p /kaggle/working/test-output && cp /kaggle/working/AlphaPose/examples/demo/*.jpg /kaggle/working/test-input
cd /kaggle/working/AlphaPose && python3 scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml --checkpoint pretrained_models/fast_res50_256x192.pth --indir /kaggle/working/test-input --save_img cd /kaggle/working/AlphaPose && python3 scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml --checkpoint pretrained_models/fast_res50_256x192.pth --indir /kaggle/working/test-input --outdir /kaggle/working/test-output --save_img
''') == 0 ''') == 0

1
deps/alpha-pose vendored Submodule

@ -0,0 +1 @@
Subproject commit e2e4394f996efd73bb8c8eb53ccfb305136ae0f6

@ -1154,3 +1154,27 @@ def kernel_15(
t23, t23,
] ]
) )
def kernel_16():
import argparse
import os
import platform
import sys
import time
import numpy as np
import torch
from tqdm import tqdm
import natsort
from detector.apis import get_detector
from trackers.tracker_api import Tracker
from trackers.tracker_cfg import cfg as tcfg
from trackers import track
from alphapose.models import builder
from alphapose.utils.config import update_config
from alphapose.utils.detector import DetectionLoader
from alphapose.utils.transforms import flip, flip_heatmap
from alphapose.utils.vis import getTime
from alphapose.utils.webcam_detector import WebCamDetectionLoader
from alphapose.utils.writer import DataWriter